diff --git a/lib/migration_generator/migration_generator.ex b/lib/migration_generator/migration_generator.ex
index ed4b16a..b5cd1dc 100644
--- a/lib/migration_generator/migration_generator.ex
+++ b/lib/migration_generator/migration_generator.ex
@@ -1002,10 +1002,13 @@ defmodule AshPostgres.MigrationGenerator do
   end
 
   defp after?(
-         %Operation.AddCheckConstraint{table: table, schema: schema},
+         %Operation.AddCheckConstraint{table: table, schema: schema, constraint: %{name: name}},
          %Operation.RemoveCheckConstraint{
            table: table,
-           schema: schema
+           schema: schema,
+           constraint: %{
+             name: name
+           }
          }
        ),
        do: true