diff --git a/dagrunner/tests/plugin_framework/test_DataPolling.py b/dagrunner/tests/plugin_framework/test_DataPolling.py
index 7280e90..20f21c9 100644
--- a/dagrunner/tests/plugin_framework/test_DataPolling.py
+++ b/dagrunner/tests/plugin_framework/test_DataPolling.py
@@ -49,4 +49,7 @@ def test_specified_host(tmp_file, capsys):
     host_tmp_file = f"{socket.gethostname()}:{tmp_file}"
     call_dp(host_tmp_file)
     captured = capsys.readouterr()
-    assert f"The following files were polled and found: ['{tmp_file}']" in captured.out
+    assert (
+        f"The following files were polled and found: ['{tmp_file}'] on "
+        f"'{host_tmp_file}'" in captured.out
+    )