diff --git a/model/__init__.py b/model/__init__.py
index 06dad07..d577c8d 100644
--- a/model/__init__.py
+++ b/model/__init__.py
@@ -1 +1 @@
-import action_conditional_video_prediction
\ No newline at end of file
+from .action_conditional_video_prediction import train
\ No newline at end of file