diff --git a/tests/integration/network/gateway_test.py b/tests/integration/network/gateway_test.py
index b58b9cc12fb..e5c1e4e6adf 100644
--- a/tests/integration/network/gateway_test.py
+++ b/tests/integration/network/gateway_test.py
@@ -39,8 +39,12 @@ def test_domain_connect_to_gateway(domain_1_port, gateway_port):
     assert proxy_domain_client.metadata == domain_client.metadata
     assert proxy_domain_client.user_role == ServiceRole.NONE
 
-    domain_client.login(email="info@openmined.org", password="changethis")
-    proxy_domain_client.login(email="info@openmined.org", password="changethis")
+    domain_client = domain_client.login(
+        email="info@openmined.org", password="changethis"
+    )
+    proxy_domain_client = proxy_domain_client.login(
+        email="info@openmined.org", password="changethis"
+    )
 
     assert proxy_domain_client.logged_in_user == "info@openmined.org"
     assert proxy_domain_client.user_role == ServiceRole.ADMIN