diff --git a/mem0/embeddings/huggingface.py b/mem0/embeddings/huggingface.py
index 7e7ca590..6b187d81 100644
--- a/mem0/embeddings/huggingface.py
+++ b/mem0/embeddings/huggingface.py
@@ -1,4 +1,4 @@
-from embedding.base import EmbeddingBase
+from mem0.embeddings.base import EmbeddingBase
 from sentence_transformers import SentenceTransformer
 
 
diff --git a/mem0/embeddings/ollama.py b/mem0/embeddings/ollama.py
index 649fe161..00289d4e 100644
--- a/mem0/embeddings/ollama.py
+++ b/mem0/embeddings/ollama.py
@@ -1,5 +1,5 @@
 import ollama
-from embedding.base import EmbeddingBase
+from mem0.embeddings.base import EmbeddingBase
 
 
 class OllamaEmbedding(EmbeddingBase):