diff --git a/src/neo4j_genai/embedder.py b/src/neo4j_genai/embedder.py
index be3e4c8f..4cd69529 100644
--- a/src/neo4j_genai/embedder.py
+++ b/src/neo4j_genai/embedder.py
@@ -14,19 +14,18 @@
 #  limitations under the License.
 
 from abc import ABC, abstractmethod
-from .types import EmbeddingVector
 
 
 class Embedder(ABC):
     """Interface for embedding models."""
 
     @abstractmethod
-    def embed_query(self, text: str) -> EmbeddingVector:
+    def embed_query(self, text: str) -> list[float]:
         """Embed query text.
 
         Args:
             text (str): Text to convert to vector embedding
 
         Returns:
-            EmbeddingVector: A vector embedding.
+            list[float]: A vector embedding.
         """
diff --git a/src/neo4j_genai/types.py b/src/neo4j_genai/types.py
index be6f6281..5747aeea 100644
--- a/src/neo4j_genai/types.py
+++ b/src/neo4j_genai/types.py
@@ -23,10 +23,6 @@ class VectorSearchRecord(BaseModel):
     score: float
 
 
-class EmbeddingVector(BaseModel):
-    vector: list[float]
-
-
 class IndexModel(BaseModel):
     driver: Any