diff --git a/integration_tests/conftest.py b/integration_tests/conftest.py
index 9a197b1244..8a617bba38 100644
--- a/integration_tests/conftest.py
+++ b/integration_tests/conftest.py
@@ -620,8 +620,10 @@ def run_claude_structured(
         "--rm",
         "--network",
         egg_stack.isolated_network,
+        "--add-host",
+        f"egg-gateway:{egg_stack.gateway_isolated_ip}",
         "-e",
-        f"EGG_GATEWAY_URL=http://{egg_stack.gateway_isolated_ip}:9848",
+        "GATEWAY_URL=http://egg-gateway:9848",
         "-e",
         f"EGG_SESSION_TOKEN={session_token}",
         "-e",