diff --git a/core/services/ModuleService.php b/core/services/ModuleService.php
index f3b4703..d2417b3 100644
--- a/core/services/ModuleService.php
+++ b/core/services/ModuleService.php
@@ -95,14 +95,19 @@ class ModuleService
             $migrationFile = basename($migrationPath);
             $migration     = str_replace('.php', '', $migrationFile);
             if (!$this->migrationExists($migration)) {
+                $addMigrationRecord = false;
                 // run migration
                 require $migrationPath;
                 $obj = new $migration;
                 if (method_exists($obj, 'safeUp')) {
                     $obj->safeUp();
-                    Yii::$app->moduleManager->appendToMigrationTable($migration);
-                } elseif (method_exists($obj, 'up')) {
+                    $addMigrationRecord = true;
+                }
+                if (method_exists($obj, 'up')) {
                     $obj->up();
+                    $addMigrationRecord = true;
+                }
+                if ($addMigrationRecord) {
                     Yii::$app->moduleManager->appendToMigrationTable($migration);
                 }
             }