diff --git a/test/test_helpers.py b/test/test_helpers.py
index 259423717f43ce26adc561e27f32271de4e9e897..7de58f331ea4b6eeaf46b6d8b10f8ad1da487f3c 100644
--- a/test/test_helpers.py
+++ b/test/test_helpers.py
@@ -5,6 +5,7 @@ import os
 import keras
 import numpy as np
 import mock
+import platform
 
 
 class TestToList:
@@ -168,7 +169,7 @@ class TestPrepareHost:
         with pytest.raises(OSError) as e:
             prepare_host()
         assert "unknown host 'NotExistingHostName'" in e.value.args[0]
-        if "runner-6HmDp9Qd-project-2411-concurrent" not in socket.gethostname():
+        if "runner-6HmDp9Qd-project-2411-concurrent" not in platform.node():
             mock_host.return_value = "linux-gzsx"
             with pytest.raises(NotADirectoryError) as e:
                 prepare_host()