diff --git a/docs/environment_setup.md b/docs/environment_setup.md
index 5365904d9..e7a4b2e7c 100644
--- a/docs/environment_setup.md
+++ b/docs/environment_setup.md
@@ -24,7 +24,7 @@ def hello_world_wf() -> str:
     return res
 
 if __name__ == "__main__":
-    print(f"Running my_wf() {my_wf()}")
+    print(f"Running my_wf() {hello_world_wf()}")
 ```
 
 To install `flytekit`, run the following command: