diff --git a/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/mixed_cluster/10_basic.yaml b/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/mixed_cluster/10_basic.yaml
index 495e1d0b3d7..62fe1c77627 100644
--- a/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/mixed_cluster/10_basic.yaml
+++ b/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/mixed_cluster/10_basic.yaml
@@ -4,11 +4,7 @@
      cluster.health:
         wait_for_status: green
 
- - do:
-      indices.get_settings:
-        index: test_index
-
- - match: { test_index.settings.index.number_of_replicas: "0" }
+ - match: { number_of_nodes: 2 }
 
  - do:
      search:
diff --git a/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/upgraded_cluster/10_basic.yaml b/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/upgraded_cluster/10_basic.yaml
index b7a38b4123b..eec2820ce32 100644
--- a/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/upgraded_cluster/10_basic.yaml
+++ b/qa/rolling-upgrade/src/test/resources/rest-api-spec/test/upgraded_cluster/10_basic.yaml
@@ -4,6 +4,8 @@
      cluster.health:
         wait_for_status: green
 
+ - match: { number_of_nodes: 2 }
+
  - do:
      search:
        index: test_index