diff --git a/framework/views/migration.php b/framework/views/migration.php
index 1a7d1f9517..584e8e4be4 100644
--- a/framework/views/migration.php
+++ b/framework/views/migration.php
@@ -9,8 +9,9 @@ echo "<?php\n";
 ?>
 
 use yii\db\Schema;
+use yii\db\Migration;
 
-class <?= $className ?> extends \yii\db\Migration
+class <?= $className ?> extends Migration
 {
     public function up()
     {