diff --git a/tests/product/test_collect.py b/tests/product/test_collect.py
index 5d74018f..d12d066a 100644
--- a/tests/product/test_collect.py
+++ b/tests/product/test_collect.py
@@ -200,9 +200,7 @@ def test_collect_system_info_server_stopped(self):
         message = '\nFatal error: [%s] Unable to access node ' \
                   'information. Please check that server is up with ' \
                   'command: server status\n\nAborting.\n'
-        expected = ''
-        for host in self.cluster.all_internal_hosts():
-            expected += message % host
+        expected = message % self.cluster.internal_master
         self.assertEqualIgnoringOrder(actual, expected)
 
     def _add_custom_log_location(self, new_log_location):