diff --git a/src/Commands/MigrationGeneratorCommand.php b/src/Commands/MigrationGeneratorCommand.php
index 002639a..8372feb 100644
--- a/src/Commands/MigrationGeneratorCommand.php
+++ b/src/Commands/MigrationGeneratorCommand.php
@@ -4,13 +4,13 @@
 
 use Illuminate\Filesystem\Filesystem;
 use Orchestra\Canvas\Core\Concerns\MigrationGenerator;
-use Orchestra\Canvas\Core\Contracts\MigrationGenerator as MigrationGeneratorContract;
+use Orchestra\Canvas\Core\Contracts\Generator;
 
 /**
  * @property string|null $name
  * @property string|null $description
  */
-abstract class MigrationGeneratorCommand extends \Illuminate\Console\MigrationGeneratorCommand implements MigrationGeneratorContract
+abstract class MigrationGeneratorCommand extends \Illuminate\Console\MigrationGeneratorCommand implements Generator
 {
     use MigrationGenerator;