diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py
index 04e8e91..2cd5dee 100644
--- a/anansyTypes/schemas.py
+++ b/anansyTypes/schemas.py
@@ -36,7 +36,7 @@ class EmbeddingRequest(ModifiedBaseModel):
     word_ids: List[int]
 
 class EmbeddingResponse(ModifiedBaseModel):
-    embeddings: List[Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]]
+    embeddings: List[Embedding]
 
 class ResponseRequest(ModifiedBaseModel):
     text: str