diff --git a/norch/utils/data/__init__.py b/norch/utils/data/__init__.py
index e5ea955..2397554 100644
--- a/norch/utils/data/__init__.py
+++ b/norch/utils/data/__init__.py
@@ -1,5 +1,4 @@
 from .dataset import *
 from .example import *
 from .dataloader import *
-from .batch import *
-from .distributed import *
\ No newline at end of file
+from .batch import *
\ No newline at end of file