diff --git a/src/steps/relay.py b/src/steps/relay.py
index 1bb5dbbb..8630c085 100644
--- a/src/steps/relay.py
+++ b/src/steps/relay.py
@@ -52,7 +52,7 @@ class StepsRelay:
         for index, peer in enumerate(peer_list):
             logger.debug(f"Checking that peer NODE_{index + 1} {peer.image} can find the published message")
             get_messages_response = peer.get_messages(pubsub_topic or self.test_pubsub_topic)
-            assert get_messages_response, f"Peer NODE_{index} {peer.image} couldn't find any messages"
+            assert get_messages_response, f"Peer NODE_{index}:{peer.image} couldn't find any messages"
             received_message = message_rpc_response_schema.load(get_messages_response[0])
             self.assert_received_message(message, received_message)
 
diff --git a/tests/relay/test_publish.py b/tests/relay/test_publish.py
index 2f893a23..6ecfa2b6 100644
--- a/tests/relay/test_publish.py
+++ b/tests/relay/test_publish.py
@@ -65,7 +65,7 @@ class TestRelayPublish(StepsRelay):
                 self.check_published_message_reaches_peer(message, message_propagation_delay=2)
                 raise AssertionError("Duplicate message was retrieved twice")
             except Exception as ex:
-                assert "Peer node couldn't find any messages" in str(ex)
+                assert "couldn't find any messages" in str(ex)
 
     def test_publish_with_valid_content_topics(self):
         failed_content_topics = []
@@ -204,7 +204,7 @@ class TestRelayPublish(StepsRelay):
             self.check_published_message_reaches_peer(message)
             raise AssertionError("Duplicate message was retrieved twice")
         except Exception as ex:
-            assert "Peer node couldn't find any messages" in str(ex)
+            assert "couldn't find any messages" in str(ex)
 
     def test_publish_while_peer_is_paused(self):
         message = self.create_message()