diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py
index f62c672..9140fe2 100644
--- a/anansyTypes/schemas.py
+++ b/anansyTypes/schemas.py
@@ -64,7 +64,7 @@ class GetWordResponse(ModifiedBaseModel):
     word: Union[dict, None]
 
 class GetSimilarWordRequest(ModifiedBaseModel):
-    embedding: Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]
+    embedding: dict
 
 class GetAllPhrasesResponse(ModifiedBaseModel):
     phrases: List[dict]