diff --git a/lotus/vector_store/weaviate_vs.py b/lotus/vector_store/weaviate_vs.py
index eab46d0..82fff0b 100644
--- a/lotus/vector_store/weaviate_vs.py
+++ b/lotus/vector_store/weaviate_vs.py
@@ -133,7 +133,7 @@ def __call__(self,
             indices=np.array(all_indices, dtype=np.int64).tolist()
         )
 
-    def get_vectors_from_index(self, collection_name: str, ids: list[uuid4]) -> NDArray[np.float64]:
+    def get_vectors_from_index(self, collection_name: str, ids: list[str]) -> NDArray[np.float64]:
         """Retrieve vectors for specific document IDs"""
         collection = self.client.collections.get(collection_name)