diff --git a/src/uagents/models.py b/src/uagents/models.py
index 703e8d9d..69b2f6eb 100644
--- a/src/uagents/models.py
+++ b/src/uagents/models.py
@@ -9,7 +9,7 @@ class Model(BaseModel):
     _schema_no_descriptions = None
 
     @staticmethod
-    def remove_descriptions(schema: Dict[str, Dict[str, str]]):
+    def remove_descriptions(schema: Dict[str, Any]):
         if not "properties" in schema:
             return