diff --git a/tests/test_integration.py b/tests/test_integration.py
index afb903a..35169ef 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -4,9 +4,14 @@
 import time
 import subprocess
 import itertools
-from ontologytimemachine.custom_proxy import IP, PORT
+from ontologytimemachine.utils.config import Config
 
 
+default_cfg: Config = Config()
+
+IP = default_cfg.port
+PORT = default_cfg.host
+
 PROXY = f"{IP}:{PORT}"
 HTTP_PROXY = f"http://{PROXY}"
 HTTPS_PROXY = f"http://{PROXY}"