diff --git a/src/toolbox_langchain_sdk/utils.py b/src/toolbox_langchain_sdk/utils.py
index 2ec8e87a..036624d6 100644
--- a/src/toolbox_langchain_sdk/utils.py
+++ b/src/toolbox_langchain_sdk/utils.py
@@ -124,7 +124,7 @@ def _parse_type(type_: str) -> Any:
         return str
     elif type_ == "integer":
         return int
-    elif type_ == "number":
+    elif type_ == "float":
         return float
     elif type_ == "boolean":
         return bool
diff --git a/tests/test_utils.py b/tests/test_utils.py
index 2b59737a..2ce6fced 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -169,7 +169,7 @@ def test_schema_to_model_empty(self):
         [
             ("string", str),
             ("integer", int),
-            ("number", float),
+            ("float", float),
             ("boolean", bool),
             ("array", list),
         ],