diff --git a/sekoia_automation/scripts/new_module/template/{{cookiecutter.module_dir}}/main.py b/sekoia_automation/scripts/new_module/template/{{cookiecutter.module_dir}}/main.py
index ffb6584..7221238 100644
--- a/sekoia_automation/scripts/new_module/template/{{cookiecutter.module_dir}}/main.py
+++ b/sekoia_automation/scripts/new_module/template/{{cookiecutter.module_dir}}/main.py
@@ -1,4 +1,4 @@
-from {{cookiecutter.module_name.lower()}}_modules import {{cookiecutter.module_name.capitalize()}}Module
+from {{cookiecutter.module_name.lower().replace(" ", "_")}}_modules import {{cookiecutter.module_name.title().replace(" ", "")}}Module
 
 if __name__ == "__main__":
     module = {{cookiecutter.module_name.title().replace(" ", "")}}Module()