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