diff --git a/pkg/sql/logictest/testdata/logic_test/distsql_tenant_locality b/pkg/sql/logictest/testdata/logic_test/distsql_tenant_locality index 18ef8e4112a3..057ada6dd873 100644 --- a/pkg/sql/logictest/testdata/logic_test/distsql_tenant_locality +++ b/pkg/sql/logictest/testdata/logic_test/distsql_tenant_locality @@ -12,8 +12,9 @@ statement ok CREATE TABLE t (k INT PRIMARY KEY, v INT, FAMILY (k, v)); INSERT INTO t SELECT i, i FROM generate_series(1, 6) AS g(i) -# Upreplicate the table's range. -statement ok +# Upreplicate the table's range. We need a retry to guarantee that the +# capability has been picked up. +statement ok retry ALTER TABLE t EXPERIMENTAL_RELOCATE VALUES (ARRAY[1, 2, 3], 0) # Split the ranges in the table. diff --git a/pkg/sql/opt/exec/execbuilder/testdata/distsql_tenant_locality b/pkg/sql/opt/exec/execbuilder/testdata/distsql_tenant_locality index ea66b6106c22..8725baf7a56c 100644 --- a/pkg/sql/opt/exec/execbuilder/testdata/distsql_tenant_locality +++ b/pkg/sql/opt/exec/execbuilder/testdata/distsql_tenant_locality @@ -12,8 +12,9 @@ user root statement ok CREATE TABLE t (k INT PRIMARY KEY, v INT, FAMILY (k, v)) -# Upreplicate the table's range. -statement ok +# Upreplicate the table's range. We need a retry to guarantee that the +# capability has been picked up. +statement ok retry ALTER TABLE t EXPERIMENTAL_RELOCATE VALUES (ARRAY[1, 2, 3], 0) # Split the ranges in the table.