diff --git a/packages/compass-aggregations/src/modules/index.ts b/packages/compass-aggregations/src/modules/index.ts
index 59686b726d0..1f5de34002b 100644
--- a/packages/compass-aggregations/src/modules/index.ts
+++ b/packages/compass-aggregations/src/modules/index.ts
@@ -95,10 +95,10 @@ const rootReducer = combineReducers({
 export type RootState = ReturnType<typeof rootReducer>;
 
 export type PipelineBuilderExtraArgs = {
-  globalAppRegistry: AppRegistry;
-  localAppRegistry: AppRegistry;
+  globalAppRegistry: Pick<AppRegistry, 'on' | 'emit' | 'removeListener'>;
+  localAppRegistry: Pick<AppRegistry, 'on' | 'emit' | 'removeListener'>;
   pipelineBuilder: PipelineBuilder;
-  pipelineStorage: PipelineStorage;
+  pipelineStorage: PipelineStorage | undefined;
   workspaces: WorkspacesService;
   preferences: PreferencesAccess;
   logger: Logger;
diff --git a/packages/compass-aggregations/src/stores/store.ts b/packages/compass-aggregations/src/stores/store.ts
index 7ee9ebedb01..cbefaaa7dae 100644
--- a/packages/compass-aggregations/src/stores/store.ts
+++ b/packages/compass-aggregations/src/stores/store.ts
@@ -135,9 +135,7 @@ export function activateAggregationsPlugin(
 
   const stagesIdAndType = mapStoreStagesToStageIdAndType(stages);
 
-  const store: Store<RootState> & {
-    dispatch: PipelineBuilderThunkDispatch;
-  } = createStore(
+  const store = createStore(
     reducer,
     {
       // TODO: move this to thunk extra arg
@@ -189,6 +187,7 @@ export function activateAggregationsPlugin(
         atlasAiService,
         connectionInfoRef,
         connectionScopedAppRegistry,
+        dataService,
       })
     )
   );