diff --git a/src/google/adk/code_executors/gke_code_executor.py b/src/google/adk/code_executors/gke_code_executor.py
index b44aa19357..fb4d160d84 100644
--- a/src/google/adk/code_executors/gke_code_executor.py
+++ b/src/google/adk/code_executors/gke_code_executor.py
@@ -29,13 +29,13 @@
 from .code_execution_utils import CodeExecutionInput
 from .code_execution_utils import CodeExecutionResult
 
-try:
-  from agentic_sandbox import SandboxClient
-except ImportError:
-  SandboxClient = None
-
 if TYPE_CHECKING:
-  from agentic_sandbox import SandboxClient
+  from k8s_agent_sandbox import SandboxClient
+else:
+  try:
+    from k8s_agent_sandbox import SandboxClient
+  except ImportError:
+    SandboxClient = None
 
 # Expose these for tests to monkeypatch.
 client = k8s.client