diff --git a/src/test/regress/expected/multi_transaction_recovery.out b/src/test/regress/expected/multi_transaction_recovery.out
index ad5f5e699..1e4cea224 100644
--- a/src/test/regress/expected/multi_transaction_recovery.out
+++ b/src/test/regress/expected/multi_transaction_recovery.out
@@ -352,7 +352,10 @@ SELECT recover_prepared_transactions();
                              0
 (1 row)
 
-SELECT shardid INTO selected_shard FROM pg_dist_shard WHERE logicalrelid='test_2pcskip'::regclass LIMIT 1;
+SELECT shardid INTO selected_shard
+FROM citus_shards
+WHERE table_name='test_2pcskip'::regclass AND nodeport = :worker_1_port
+LIMIT 1;
 SELECT COUNT(*) FROM pg_dist_transaction;
  count
 ---------------------------------------------------------------------
diff --git a/src/test/regress/sql/multi_transaction_recovery.sql b/src/test/regress/sql/multi_transaction_recovery.sql
index 5156c83e3..5b5afb2e2 100644
--- a/src/test/regress/sql/multi_transaction_recovery.sql
+++ b/src/test/regress/sql/multi_transaction_recovery.sql
@@ -193,7 +193,11 @@ SELECT create_distributed_table('test_2pcskip', 'a');
 INSERT INTO test_2pcskip SELECT i FROM generate_series(0, 5)i;
 SELECT recover_prepared_transactions();
 
-SELECT shardid INTO selected_shard FROM pg_dist_shard WHERE logicalrelid='test_2pcskip'::regclass LIMIT 1;
+SELECT shardid INTO selected_shard
+FROM citus_shards
+WHERE table_name='test_2pcskip'::regclass AND nodeport = :worker_1_port
+LIMIT 1;
+
 SELECT COUNT(*) FROM pg_dist_transaction;
 BEGIN;
 SET LOCAL citus.defer_drop_after_shard_move TO OFF;