diff --git a/src/crewai/cli/templates/flow/main.py b/src/crewai/cli/templates/flow/main.py
index 38d2d8736..f6cbc2f35 100644
--- a/src/crewai/cli/templates/flow/main.py
+++ b/src/crewai/cli/templates/flow/main.py
@@ -37,29 +37,14 @@ class PoemFlow(Flow[PoemState]):
             f.write(self.state.poem)
         print(f"State after save_poem: {self.state}")
 
-async def run_flow():
-    """
-    Run the flow.
-    """
+async def kickoff():
     poem_flow = PoemFlow()
     await poem_flow.kickoff()
 
-async def plot_flow():
-    """
-    Plot the flow.
-    """
+def plot():
     poem_flow = PoemFlow()
     poem_flow.plot()
 
 
-def main():
-    asyncio.run(run_flow())
-
-
-def plot():
-    asyncio.run(plot_flow())
-
-
-
 if __name__ == "__main__":
-    main()
+    asyncio.run(kickoff())
diff --git a/src/crewai/cli/templates/flow/pyproject.toml b/src/crewai/cli/templates/flow/pyproject.toml
index 490d66346..a9c0dd916 100644
--- a/src/crewai/cli/templates/flow/pyproject.toml
+++ b/src/crewai/cli/templates/flow/pyproject.toml
@@ -10,8 +10,8 @@ dependencies = [
 ]
 
 [project.scripts]
-{{folder_name}} = "{{folder_name}}.main:main"
-run_flow = "{{folder_name}}.main:main"
+{{folder_name}} = "{{folder_name}}.main:kickoff"
+run_flow = "{{folder_name}}.main:kickoff"
 plot_flow = "{{folder_name}}.main:plot"
 
 [build-system]