diff --git a/src/command/stubs/migrate.stub b/src/command/stubs/migrate.stub
index 51ac0c7..88d16e2 100644
--- a/src/command/stubs/migrate.stub
+++ b/src/command/stubs/migrate.stub
@@ -1,6 +1,7 @@
 <?php
 
 use think\migration\Migrator;
+use think\migration\db\Column;
 
 class $className extends Migrator
 {