diff --git a/cli.go b/cli.go
index e655e0a..54e2177 100644
--- a/cli.go
+++ b/cli.go
@@ -155,10 +155,11 @@ func (c *Cli) CreateCommand(ctx context.Context) *cobra.Command {
 	var createCmd = &cobra.Command{
 		Use:   "create",
 		Short: "Create a new migration",
-		Args:  cobra.ExactArgs(1),
 		Run: func(cmd *cobra.Command, args []string) {
-			migrationName := args[0]
-			err := c.migration.Create(migrationName)
+			dir, _ := cmd.Flags().GetString("dir")
+			name, _ := cmd.Flags().GetString("name")
+
+			err := c.migration.SetMigrationFilesDir(dir).Create(name)
 			if err != nil {
 				log.Println("Error creating migration:", err)
 				return
@@ -166,6 +167,11 @@ func (c *Cli) CreateCommand(ctx context.Context) *cobra.Command {
 		},
 	}
 
+	createCmd.Flags().StringP("name", "n", "", "name of the migration")
+	createCmd.Flags().StringP("dir", "d", "", "directory of the migration")
+	createCmd.MarkFlagRequired("name")
+	createCmd.MarkFlagRequired("dir")
+
 	return createCmd
 }