diff --git a/backend/apps/webui/routers/tools.py b/backend/apps/webui/routers/tools.py
index 900e17999..3ab75187a 100644
--- a/backend/apps/webui/routers/tools.py
+++ b/backend/apps/webui/routers/tools.py
@@ -164,9 +164,9 @@ async def get_toolkit_valves_spec_by_id(
             toolkit_module = load_toolkit_module_by_id(id)
             request.app.state.TOOLS[id] = toolkit_module
 
-        if hasattr(toolkit_module, "UserValves"):
-            UserValves = toolkit_module.UserValves
-            return UserValves.schema()
+        if hasattr(toolkit_module, "Valves"):
+            Valves = toolkit_module.Valves
+            return Valves.schema()
         return None
     else:
         raise HTTPException(