diff --git a/examples/tools/shell.py b/examples/tools/shell.py
index 6fa97af3b0..b4ed4becd4 100644
--- a/examples/tools/shell.py
+++ b/examples/tools/shell.py
@@ -11,13 +11,13 @@
     ShellCallOutcome,
     ShellCommandOutput,
     ShellCommandRequest,
+    ShellOnApprovalFunctionResult,
     ShellResult,
     ShellTool,
     trace,
 )
 from agents.items import ToolApprovalItem
 from agents.run_context import RunContextWrapper
-from agents.tool import ShellOnApprovalFunctionResult
 
 SHELL_AUTO_APPROVE = os.environ.get("SHELL_AUTO_APPROVE") == "1"
 
diff --git a/src/agents/__init__.py b/src/agents/__init__.py
index ce2f8fbca8..f345edfa2b 100644
--- a/src/agents/__init__.py
+++ b/src/agents/__init__.py
@@ -150,6 +150,8 @@
     ShellCommandOutput,
     ShellCommandRequest,
     ShellExecutor,
+    ShellOnApprovalFunction,
+    ShellOnApprovalFunctionResult,
     ShellResult,
     ShellTool,
     ShellToolContainerAutoEnvironment,
@@ -474,6 +476,8 @@ def enable_verbose_stdout_logging():
     "ShellToolHostedEnvironment",
     "ShellToolEnvironment",
     "ShellExecutor",
+    "ShellOnApprovalFunction",
+    "ShellOnApprovalFunctionResult",
     "ShellResult",
     "ShellTool",
     "ApplyPatchEditor",