diff --git a/src/lfx/src/lfx/components/logic/tool_router.py b/src/lfx/src/lfx/components/logic/tool_router.py
index f013ce0a77..d3db0b4fc1 100644
--- a/src/lfx/src/lfx/components/logic/tool_router.py
+++ b/src/lfx/src/lfx/components/logic/tool_router.py
@@ -1,7 +1,8 @@
 from typing import Any
 
 from lfx.custom import Component
-from lfx.io import BoolInput, HandleInput, MessageInput, MessageTextInput, MultilineInput, Output
+from lfx.io import BoolInput, MessageInput, MessageTextInput, MultilineInput, Output
+from lfx.inputs import ToolsInput
 from lfx.schema.message import Message
 
 
@@ -30,14 +31,11 @@ class ToolRouterComponent(Component):
             info="The primary text input for tool selection.",
             required=True,
         ),
-        HandleInput(
+        ToolsInput(
             name="tools",
             display_name="Tools",
-            input_types=["Tool"],
-            is_list=True,
-            required=True,
             info="Tools that can be selected based on the input categorization.",
-            real_time_refresh=True,
+            required=True,
         ),
         MessageInput(
             name="message",