diff --git a/monoloco/__init__.py b/monoloco/__init__.py
index 91341c0..fdd0fe7 100644
--- a/monoloco/__init__.py
+++ b/monoloco/__init__.py
@@ -2,4 +2,3 @@
 """Open implementation of MonoLoco."""
 
 __version__ = '0.4.3'
-
diff --git a/monoloco/run.py b/monoloco/run.py
index 80ba6a4..a237672 100644
--- a/monoloco/run.py
+++ b/monoloco/run.py
@@ -100,8 +100,8 @@ def main():
             from .visuals.webcam import webcam
             webcam(args)
         else:
-            from predict import predict
-            predict(args)
+            from . import predict
+            predict.predict(args)
 
     elif args.command == 'prep':
         if 'nuscenes' in args.dataset: