Fix unit tests

Author: pchila

.mockery.yaml

@@ -8,9 +8,6 @@ packages:
   github.com/elastic/elastic-agent/internal/pkg/agent/application:
     interfaces:
       rollbacksSource: {}
-  github.com/elastic/elastic-agent/internal/pkg/agent/application/gateway/fleet:
-    interfaces:
-      rollbacksSource: {}
   github.com/elastic/elastic-agent/internal/pkg/agent/application/actions/handlers:
     interfaces:
       Uploader: {}
@@ -20,6 +17,9 @@ packages:
   github.com/elastic/elastic-agent/internal/pkg/agent/application/info:
     interfaces:
       Agent: {}
+  github.com/elastic/elastic-agent/internal/pkg/agent/application/gateway/fleet:
+    interfaces:
+      rollbacksSource: { }
   github.com/elastic/elastic-agent/internal/pkg/agent/application/upgrade:
     interfaces:
       WatcherHelper: {}