diff --git a/integration_test/support/migration.exs b/integration_test/support/migration.exs
index db890cb7..68815ed7 100644
--- a/integration_test/support/migration.exs
+++ b/integration_test/support/migration.exs
@@ -111,10 +111,10 @@ defmodule Ecto.Integration.Migration do
     end
 
     create table(:composite_pk_composite_pk, primary_key: false) do
-      add :b_1, :integer
       add :a_1, references(:composite_pk, column: :a, with: [b_1: :b], type: :integer)
-      add :b_2, :integer
+      add :b_1, :integer
       add :a_2, references(:composite_pk, column: :a, with: [b_2: :b], type: :integer)
+      add :b_2, :integer
     end
 
 
@@ -122,13 +122,13 @@ defmodule Ecto.Integration.Migration do
       modify :composite_a, references(:composite_pk, column: :a, with: [composite_b: :b], type: :integer)
     end
 
-    create table(:posts_composite_pk) do
-      add :post_id, references(:posts), primary_key: true
-      add :composite_a, references(:composite_pk, column: :a, with: [composite_b: :b], type: :integer), primary_key: true
-      add :composite_b, :integer, primary_key: true
+    create table(:one_to_one_composite_pk) do
+      add :composite_a, references(:composite_pk, column: :a, with: [composite_b: :b], type: :integer)
+      add :composite_b, :integer
+      timestamps()
     end
 
-    create unique_index(:posts_composite_pk, [:post_id, :composite_a, :composite_b])
+    create unique_index(:one_to_one_composite_pk, [:composite_a, :composite_b])
 
     create table(:corrupted_pk, primary_key: false) do
       add :a, :string