diff --git a/pyproject.toml b/pyproject.toml
index 5bce68f..3d23f01 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ click = "^8.1.3"
 [tool.poetry.dev-dependencies]
 
 [tool.poetry.scripts]
-update_django_environ_sample = "update_django_environ_sample"
+update_django_environ_sample = "update_django_environ_sample:main"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
diff --git a/update_django_environ_sample/__init__.py b/update_django_environ_sample/__init__.py
index fad315e..ada4a0f 100644
--- a/update_django_environ_sample/__init__.py
+++ b/update_django_environ_sample/__init__.py
@@ -57,5 +57,9 @@ def generate_env_sample(*, environment_defaults_path: str, output: str, defaults
         f.write(text)
 
 
-if __name__ == "__main__":
+def main():
     generate_env_sample()
+
+
+if __name__ == "__main__":
+    main()