diff --git a/lmm_tools/__init__.py b/lmm_tools/__init__.py
index 75e4f900..e69de29b 100644
--- a/lmm_tools/__init__.py
+++ b/lmm_tools/__init__.py
@@ -1,3 +0,0 @@
-from .data import Data, get_data
-from .lmm import LMM, get_lmm
-from .emb import Embedder, get_embedder
diff --git a/lmm_tools/data/__init__.py b/lmm_tools/data/__init__.py
index 5d816078..050da913 100644
--- a/lmm_tools/data/__init__.py
+++ b/lmm_tools/data/__init__.py
@@ -1 +1 @@
-from .data import Data, build_data
+from .data import DataStore, build_data_store