diff --git a/tools/my-plugin/src/generators/my-generator/generator.spec.ts b/tools/my-plugin/src/generators/my-generator/generator.spec.ts
index 1989922..30e6824 100644
--- a/tools/my-plugin/src/generators/my-generator/generator.spec.ts
+++ b/tools/my-plugin/src/generators/my-generator/generator.spec.ts
@@ -17,4 +17,9 @@ describe('my-generator generator', () => {
     const config = readProjectConfiguration(tree, 'test');
     expect(config).toBeDefined();
   });
+
+  it('should create workflow file', async () => {
+    await myGeneratorGenerator(tree, options);
+    expect(tree.exists('test/src/app/app.workflow.ts')).toBeTruthy();
+  });
 });
diff --git a/tools/my-plugin/src/generators/my-step/generator.spec.ts b/tools/my-plugin/src/generators/my-step/generator.spec.ts
deleted file mode 100644
index c398745..0000000
--- a/tools/my-plugin/src/generators/my-step/generator.spec.ts
+++ /dev/null
@@ -1,20 +0,0 @@
-import { createTreeWithEmptyWorkspace } from '@nx/devkit/testing';
-import { Tree, readProjectConfiguration } from '@nx/devkit';
-
-import { myStepGenerator } from './generator';
-import { MyStepGeneratorSchema } from './schema';
-
-describe('my-step generator', () => {
-  let tree: Tree;
-  const options: MyStepGeneratorSchema = { name: 'test' };
-
-  beforeEach(() => {
-    tree = createTreeWithEmptyWorkspace();
-  });
-
-  it('should run successfully', async () => {
-    await myStepGenerator(tree, options);
-    const config = readProjectConfiguration(tree, 'test');
-    expect(config).toBeDefined();
-  });
-});