diff --git a/packages/codemod/transformations/core/v2-to-v3/optional/__tests__/Enums-migration.test.ts b/packages/codemod/transformations/core/v2-to-v3/optional/__tests__/Enums-migration.test.ts
index a8553ace7f..1c70d5cead 100644
--- a/packages/codemod/transformations/core/v2-to-v3/optional/__tests__/Enums-migration.test.ts
+++ b/packages/codemod/transformations/core/v2-to-v3/optional/__tests__/Enums-migration.test.ts
@@ -3,7 +3,7 @@ import { defineInlineTest } from "jscodeshift/src/testUtils";
 
 function prependImport(source: string): string {
   return `
-    import { Button } from "monday-ui-react-core";
+    import { Button } from "@vibe/core";
     ${source}
   `;
 }