diff --git a/__init__.py b/__init__.py
index 2702f3dbe8b63634574cf812d953db28b1053f4c..7b9dbff7a768970a85a2f32dc3c5e0bfe19c0f2b 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,3 +1,3 @@
-from toolkit import ClassificationProject
-from compare import overlay_ROC, overlay_loss
+from .toolkit import ClassificationProject
+from .compare import overlay_ROC, overlay_loss