diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index ad53345..deb2e47 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -49,8 +49,8 @@ jobs:
 
       - name: Run cypress-codegen
         run: |
-          bun src/cli.ts --testingType component
-          bun src/cli.ts --testingType e2e
+          bun dist/cli.mjs --testingType component
+          bun dist/cli.mjs --testingType e2e
 
       - name: Check generated types match git
         run: |
diff --git a/src/generate-content-with-exports.ts b/src/generate-content-with-exports.ts
index 5c96d22..44262af 100644
--- a/src/generate-content-with-exports.ts
+++ b/src/generate-content-with-exports.ts
@@ -12,7 +12,7 @@ limitations under the License.
 */
 
 import * as t from "@babel/types";
-import generate from "@babel/generator";
+import { generate } from "@babel/generator";
 import { format, Options } from "prettier";
 import { join, parse, relative } from "path";
 
diff --git a/src/generate-content-with-imports.ts b/src/generate-content-with-imports.ts
index 22622fb..7c70dd8 100644
--- a/src/generate-content-with-imports.ts
+++ b/src/generate-content-with-imports.ts
@@ -13,7 +13,7 @@ limitations under the License.
 
 import { parse } from "@babel/parser";
 import * as t from "@babel/types";
-import generate from "@babel/generator";
+import { generate } from "@babel/generator";
 import { readFileSync } from "fs";
 import { resolve } from "path";
 import { format, Options } from "prettier";
diff --git a/src/generate-contents-with-interface.ts b/src/generate-contents-with-interface.ts
index 33e9091..bba3c1c 100644
--- a/src/generate-contents-with-interface.ts
+++ b/src/generate-contents-with-interface.ts
@@ -24,7 +24,7 @@ import type {
   VariableDeclaration,
 } from "@babel/types";
 import * as t from "@babel/types";
-import generate from "@babel/generator";
+import { generate } from "@babel/generator";
 import { readFileSync } from "fs";
 import { resolve } from "path";
 import { format, Options } from "prettier";