diff --git a/src/test/regress/expected/isolation_distributed_transaction_id.out b/src/test/regress/expected/isolation_distributed_transaction_id.out
index 7188d948f..5fc12be0b 100644
--- a/src/test/regress/expected/isolation_distributed_transaction_id.out
+++ b/src/test/regress/expected/isolation_distributed_transaction_id.out
@@ -77,7 +77,7 @@ step s4-get-all-transactions:
 initiator_node_identifiertransaction_numbertransaction_stamp
 
 
-starting permutation: s1-create-table s1-begin s1-insert s1-get-current-transaction-id s2-get-first-worker-active-transactions
+starting permutation: s1-create-table s1-begin s1-insert s1-get-current-transaction-id s2-get-first-worker-active-transactions s1-commit
 step s1-create-table: 
 	-- some tests also use distributed table
 	CREATE TABLE distributed_transaction_id_table(some_value int, other_value int);
@@ -110,3 +110,35 @@ step s2-get-first-worker-active-transactions:
 nodename       nodeport       success        result         
 
 localhost      57637          t              (0,289)        
+step s1-commit: 
+    COMMIT;
+
+
+starting permutation: s1-begin s1-assign-transaction-id s1-has-transaction-number s2-vacuum s1-has-transaction-number s1-commit
+step s1-begin: 
+    BEGIN;
+
+step s1-assign-transaction-id: 
+    SELECT assign_distributed_transaction_id(1, 1, '2015-01-01 00:00:00+0');
+
+assign_distributed_transaction_id
+
+               
+step s1-has-transaction-number: 
+    SELECT transaction_number > 0 FROM get_current_transaction_id();
+
+?column?       
+
+t              
+step s2-vacuum: 
+    VACUUM FULL pg_dist_partition;
+
+step s1-has-transaction-number: 
+    SELECT transaction_number > 0 FROM get_current_transaction_id();
+
+?column?       
+
+t              
+step s1-commit: 
+    COMMIT;
+
diff --git a/src/test/regress/specs/isolation_distributed_transaction_id.spec b/src/test/regress/specs/isolation_distributed_transaction_id.spec
index 8d04b17f9..7625d3266 100644
--- a/src/test/regress/specs/isolation_distributed_transaction_id.spec
+++ b/src/test/regress/specs/isolation_distributed_transaction_id.spec
@@ -22,6 +22,11 @@ step "s1-assign-transaction-id"
     SELECT assign_distributed_transaction_id(1, 1, '2015-01-01 00:00:00+0');
 }
 
+step "s1-has-transaction-number"
+{
+    SELECT transaction_number > 0 FROM get_current_transaction_id();
+}
+
 step "s1-commit"
 {
     COMMIT;
@@ -57,6 +62,11 @@ step "s2-assign-transaction-id"
     SELECT assign_distributed_transaction_id(2, 2, '2015-01-02 00:00:00+0');
 }
 
+step "s2-vacuum"
+{
+    VACUUM FULL pg_dist_partition;
+}
+
 step "s2-commit"
 {
     COMMIT;
@@ -103,5 +113,7 @@ permutation "s1-begin" "s1-assign-transaction-id" "s4-get-all-transactions" "s2-
 
 # now show that distributed transaction id on the coordinator
 # is the same with the one on the worker
-permutation "s1-create-table" "s1-begin" "s1-insert" "s1-get-current-transaction-id" "s2-get-first-worker-active-transactions"
+permutation "s1-create-table" "s1-begin" "s1-insert" "s1-get-current-transaction-id" "s2-get-first-worker-active-transactions" "s1-commit"
 
+# we would initially forget the distributed transaction ID on pg_dist_partition invalidations
+permutation "s1-begin" "s1-assign-transaction-id" "s1-has-transaction-number" "s2-vacuum" "s1-has-transaction-number" "s1-commit"