diff --git a/apps/dokploy/server/server.ts b/apps/dokploy/server/server.ts
index 4b41c4dc..9d1794a3 100644
--- a/apps/dokploy/server/server.ts
+++ b/apps/dokploy/server/server.ts
@@ -62,8 +62,8 @@ void app.prepare().then(async () => {
 			await sendDokployRestartNotifications();
 		}
 
-		if (IS_CLOUD) {
-			// await migration(); //TODO: Fix migration
+		if (IS_CLOUD && process.env.NODE_ENV === "production") {
+			await migration();
 		}
 
 		server.listen(PORT);