diff --git a/README.md b/README.md
index 1b94974..5408c5b 100644
--- a/README.md
+++ b/README.md
@@ -25,7 +25,7 @@ python src/main.py fit
 
 Start inference on images:
 ```bash
-python src/main predict --ckpt_path <path_to_checkpoint>
+python src/main.py predict --ckpt_path <path_to_checkpoint>
 ```
 
 ## License
diff --git a/src/datamodule/__init__.py b/src/datamodule/__init__.py
index ef389a4..a25dfa6 100644
--- a/src/datamodule/__init__.py
+++ b/src/datamodule/__init__.py
@@ -1,2 +1,2 @@
-from DETR import DETRDataModule
-from FasterRCNN import FasterRCNNDataModule
+from .DETR import DETRDataModule
+from .FasterRCNN import FasterRCNNDataModule
diff --git a/src/module/__init__.py b/src/module/__init__.py
index cc085db..1155212 100644
--- a/src/module/__init__.py
+++ b/src/module/__init__.py
@@ -1,2 +1,2 @@
-from DETR import DETR
-from FasterRCNN import FasterRCNN
+from .DETR import DETR
+from .FasterRCNN import FasterRCNN