diff --git a/expetator/add_energy b/expetator/add_energy.py
similarity index 100%
rename from expetator/add_energy
rename to expetator/add_energy.py
diff --git a/setup.py b/setup.py
index ecb626b8a4fb6fd8974c83580408099eb27b6593..602c842a00698ee583332e997ab0c3a90751c8dd 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,7 @@ setuptools.setup(
             'remove_watermark = expetator.remove_watermark:main',
             'csv_plot = expetator.monitoring_csv:show_csv_main',
             'list_plot = expetator.monitoring_list:show_list_main',
+            'add_energy = expetator.add_energy:main',
         ]
     }