diff --git a/src/database/data-source.ts b/src/database/data-source.ts
index 198f4350..655d6386 100644
--- a/src/database/data-source.ts
+++ b/src/database/data-source.ts
@@ -47,9 +47,13 @@ export const AppDataSource = new DataSource({
   logging: true,
   //   timezone: "Z",
   entities:
-    process.env.NODE_ENV !== "production" ? ["src/entities/**/*.ts"] : ["dist/entities/**/*{.ts,.js}"],
+    process.env.NODE_ENV !== "production"
+      ? ["src/entities/**/*.ts"]
+      : ["dist/entities/**/*{.ts,.js}"],
   migrations:
-    process.env.NODE_ENV !== "production" ? ["src/entities/*.ts"] : ["dist/entities/*{.ts,.js}"],
+    process.env.NODE_ENV !== "production"
+      ? ["src/migration/**/*.ts"]
+      : ["dist/migration/**/*{.ts,.js}"],
   subscribers: [],
   logger: new MyCustomLogger(),
 });