diff --git a/README.md b/README.md
index b8b7d059a20529f873c64626be8d04c917ad34bb..246b1867768743edfffddc5fa9ea36179d01f961 100644
--- a/README.md
+++ b/README.md
@@ -19,11 +19,11 @@ poetry install
 Once the project has been installed, you can run it with the following command:
 
 ```
-poetry run flashtrain
+python -m hydra_flash.train
 ```
 
 You can, for example, change the learning rate using : 
 
 ```
-poetry run flashtrain model.learning_rate=1e-2
+python -m hydra_flash.train model.learning_rate=1e-2
 ```
diff --git a/hydra_flash/train.py b/hydra_flash/train.py
index 8af9f8157d372f0a40179d1c771df5b8b383aad7..fa5aa5faa13d7b59efbd505bc009b016e8fb0291 100644
--- a/hydra_flash/train.py
+++ b/hydra_flash/train.py
@@ -23,3 +23,7 @@ def main(cfg):
     predict_datamodule = instantiate(cfg.predict_datamodule)
     predictions = trainer.predict(model, datamodule=predict_datamodule, output="labels")
     log.info(f"predictions : {predictions}")
+
+
+if __name__ == "__main__":
+    main()