diff --git a/tests/e2e/load/conftest.py b/tests/e2e/load/conftest.py
index e2d135092fb..09fbdd06f0f 100644
--- a/tests/e2e/load/conftest.py
+++ b/tests/e2e/load/conftest.py
@@ -5,7 +5,7 @@ from collections.abc import Iterator
 import pytest
 from requests import RequestException
 
-from e2e_gateway import Gateway
+from proxy_client import ProxyClient
 from e2e_http import NoBody, Success
 from load_client import LoadClient, build_client
 from load_constants import LOAD_MODEL
@@ -23,7 +23,7 @@ def client() -> LoadClient:
     return build_client()
 
 
-def _model_is_servable(gateway: Gateway, model_name: str) -> bool:
+def _model_is_servable(gateway: ProxyClient, model_name: str) -> bool:
     result = gateway.transport.get(
         "/v1/models",
         headers=gateway.transport.master,
diff --git a/tests/e2e/load/load_client.py b/tests/e2e/load/load_client.py
index df7c91fadf9..ab2ffb8040b 100644
--- a/tests/e2e/load/load_client.py
+++ b/tests/e2e/load/load_client.py
@@ -2,13 +2,13 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 
-from e2e_gateway import Gateway, build_gateway
+from proxy_client import ProxyClient, build_proxy_client
 
 
 @dataclass(frozen=True, slots=True)
 class LoadClient:
-    gateway: Gateway
+    gateway: ProxyClient
 
 
 def build_client() -> LoadClient:
-    return LoadClient(gateway=build_gateway())
+    return LoadClient(gateway=build_proxy_client())