nit

Author: kpamaran

packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx

@@ -168,7 +168,7 @@ describe('MockDataGeneratorModal', () => {
       expect(mockServices.atlasAiService.getAbortSignalReceived()).to.be.true;
     });
 
-    it('cancels in-flight faker mapping requests when the back button is clicked', async () => {
+    it('cancels in-flight faker mapping requests when the back button is clicked after schema confirmation', async () => {
       const mockServices = createMockServicesWithSlowAiRequest();
       renderModal({ mockServices: mockServices as any });
 

packages/compass-collection/src/modules/collection-tab.ts

@@ -384,12 +384,7 @@ const reducer: Reducer<CollectionState, Action> = (
   ) {
     if (
       state.mockDataGenerator.currentStep !==
-      MockDataGeneratorStep.SCHEMA_CONFIRMATION
-    ) {
-      return state;
-    }
-
-    if (
+        MockDataGeneratorStep.SCHEMA_CONFIRMATION ||
       state.fakerSchemaGeneration.status === 'in-progress' ||
       state.fakerSchemaGeneration.status === 'completed'
     ) {