diff --git a/test/sql/updates/post.v2.sql b/test/sql/updates/post.v2.sql
index dabe4c944..77f903667 100644
--- a/test/sql/updates/post.v2.sql
+++ b/test/sql/updates/post.v2.sql
@@ -37,8 +37,8 @@ SELECT index_name FROM _timescaledb_catalog.chunk_index ORDER BY index_name;
 -- INSERT data to create a new chunk after update or restore.
 INSERT INTO devices(id,floor) VALUES
 ('dev5', 5);
-INSERT INTO "two_Partitions"("timeCustom", device_id, series_0, series_1, series_2) VALUES
-(1258894000000000000, 'dev5', 2.2, 1, 2);
+INSERT INTO "two_Partitions"("timeCustom", device_id, device_id_2, series_0, series_1, series_2) VALUES
+(1258894000000000000, 'dev5', 'dev1', 2.2, 1, 2);
 
 SELECT * FROM public."two_Partitions";
 
diff --git a/test/sql/updates/setup.v2.sql b/test/sql/updates/setup.v2.sql
index 64fbe48b4..8a57f0982 100644
--- a/test/sql/updates/setup.v2.sql
+++ b/test/sql/updates/setup.v2.sql
@@ -18,6 +18,12 @@ CREATE TABLE PUBLIC."two_Partitions" (
   PRIMARY KEY("timeCustom", device_id),
   UNIQUE("timeCustom", device_id, series_2)
 );
+
+ALTER TABLE  "two_Partitions" ADD COLUMN device_id_2 TEXT NOT NULL;
+
+ALTER TABLE "two_Partitions" ADD CONSTRAINT two_Partitions_device_id_2_fkey
+FOREIGN KEY (device_id_2) REFERENCES devices(id);
+
 CREATE INDEX ON PUBLIC."two_Partitions" (device_id, "timeCustom" DESC NULLS LAST) WHERE device_id IS NOT NULL;
 CREATE INDEX ON PUBLIC."two_Partitions" ("timeCustom" DESC NULLS LAST, series_0) WHERE series_0 IS NOT NULL;
 CREATE INDEX ON PUBLIC."two_Partitions" ("timeCustom" DESC NULLS LAST, series_1)  WHERE series_1 IS NOT NULL;
@@ -32,14 +38,14 @@ INSERT INTO devices(id,floor) VALUES
 ('dev2', 2),
 ('dev3', 3);
 
-INSERT INTO public."two_Partitions"("timeCustom", device_id, series_0, series_1, series_2) VALUES
-(1257987600000000000, 'dev1', 1.5, 2, 2),
-(1257894000000000000, 'dev2', 1.5, 1, 3),
-(1257987600000000000, 'dev3', 1.5, 1, 1),
-(1257894002000000000, 'dev1', 2.5, 3, 4);
+INSERT INTO public."two_Partitions"("timeCustom", device_id, device_id_2, series_0, series_1, series_2) VALUES
+(1257987600000000000, 'dev1', 'dev2', 1.5, 2, 2),
+(1257894000000000000, 'dev2', 'dev2', 1.5, 1, 3),
+(1257987600000000000, 'dev3', 'dev2', 1.5, 1, 1),
+(1257894002000000000, 'dev1', 'dev2', 2.5, 3, 4);
 
-INSERT INTO "two_Partitions"("timeCustom", device_id, series_0, series_1, series_2) VALUES
-(1257894100000000000, 'dev2', 1.5, 2, 6);
+INSERT INTO "two_Partitions"("timeCustom", device_id, device_id_2, series_0, series_1, series_2) VALUES
+(1257894100000000000, 'dev2', 'dev2', 1.5, 2, 6);
 
 CREATE TABLE PUBLIC.hyper_timestamp (
   time timestamp NOT NULL,