diff --git a/datastew/__init__.py b/datastew/__init__.py
index 7d38816..c408349 100644
--- a/datastew/__init__.py
+++ b/datastew/__init__.py
@@ -4,7 +4,7 @@
 
 # Importing submodules to expose their attributes if needed
 from .process import mapping, parsing
-from .repository import model, sqllite, base
+from .repository import model, sqllite, base, weaviate
 
 __all__ = [
     "mapping",
@@ -12,8 +12,8 @@
     "model",
     "base",
     "sqllite",
+    "weaviate",
     "DataDictionarySource",
-    "BaseRepository",
     "MPNetAdapter",
     "Terminology",
     "Concept",
diff --git a/datastew/repository/weaviate.py b/datastew/repository/weaviate.py
index 7fc8b2a..d490f72 100644
--- a/datastew/repository/weaviate.py
+++ b/datastew/repository/weaviate.py
@@ -6,8 +6,8 @@
 
 from weaviate.embedded import EmbeddedOptions
 
-from datastew import BaseRepository
 from datastew.repository import Mapping, Terminology, Concept
+from datastew.repository.base import BaseRepository
 from datastew.repository.weaviate_schema import terminology_schema, concept_schema, mapping_schema