diff --git a/lib/migration_generator/migration_generator.ex b/lib/migration_generator/migration_generator.ex
index 385492a..caf0c1b 100644
--- a/lib/migration_generator/migration_generator.ex
+++ b/lib/migration_generator/migration_generator.ex
@@ -1034,7 +1034,10 @@ defmodule AshPostgres.MigrationGenerator do
     true
   end
 
-  defp after?(%Operation.AlterAttribute{new_attribute: %{references: references}}, _)
+  defp after?(
+         %Operation.AlterAttribute{new_attribute: %{references: references}, table: table},
+         %{table: table}
+       )
        when not is_nil(references),
        do: true