diff --git a/docs/concepts/flows.mdx b/docs/concepts/flows.mdx
index 9e2c20230..1c8ba8b27 100644
--- a/docs/concepts/flows.mdx
+++ b/docs/concepts/flows.mdx
@@ -562,13 +562,18 @@ async def kickoff():
     poem_flow = PoemFlow()
     await poem_flow.kickoff()
 
+
 def plot():
     poem_flow = PoemFlow()
     poem_flow.plot()
 
 
-if __name__ == "__main__":
+def main():
     asyncio.run(kickoff())
+
+
+if __name__ == "__main__":
+    main()
 ```
 
 In this example, the `PoemFlow` class defines a flow that generates a sentence count, uses the `PoemCrew` to generate a poem, and then saves the poem to a file. The flow is kicked off by calling the `kickoff()` method.
@@ -669,4 +674,4 @@ Also, check out our YouTube video on how to use flows in CrewAI below!
   allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share"
   referrerpolicy="strict-origin-when-cross-origin"
   allowfullscreen
-></iframe>{" "}
+></iframe>
diff --git a/src/crewai/cli/templates/flow/main.py b/src/crewai/cli/templates/flow/main.py
index 013081769..2574e9384 100644
--- a/src/crewai/cli/templates/flow/main.py
+++ b/src/crewai/cli/templates/flow/main.py
@@ -36,10 +36,15 @@ async def kickoff():
     poem_flow = PoemFlow()
     await poem_flow.kickoff()
 
+
 def plot():
     poem_flow = PoemFlow()
     poem_flow.plot()
 
 
-if __name__ == "__main__":
+def main():
     asyncio.run(kickoff())
+
+
+if __name__ == "__main__":
+    main()
diff --git a/src/crewai/cli/templates/flow/pyproject.toml b/src/crewai/cli/templates/flow/pyproject.toml
index 2f2b4e49d..15df6b06d 100644
--- a/src/crewai/cli/templates/flow/pyproject.toml
+++ b/src/crewai/cli/templates/flow/pyproject.toml
@@ -10,9 +10,9 @@ dependencies = [
 ]
 
 [project.scripts]
-{{folder_name}} = "{{folder_name}}.main:kickoff"
-kickoff = "{{folder_name}}.main:kickoff"
-plot = "{{folder_name}}.main:plot"
+flows_kickoff_cli = "flows_kickoff_cli.main:main"
+kickoff = "flows_kickoff_cli.main:main"
+plot = "flows_kickoff_cli.main:plot"
 
 [build-system]
 requires = ["hatchling"]