diff --git a/cookbook/pocketflow-thinking/flow.py b/cookbook/pocketflow-thinking/flow.py
index bef6893..3ec7b87 100644
--- a/cookbook/pocketflow-thinking/flow.py
+++ b/cookbook/pocketflow-thinking/flow.py
@@ -3,8 +3,8 @@ from nodes import ChainOfThoughtNode
 
 def create_chain_of_thought_flow():
     # Create a ChainOfThoughtNode
-    cot_node = ChainOfThoughtNode()
-    # max_retries=3, wait=10
+    cot_node = ChainOfThoughtNode(max_retries=3, wait=10)
+    
     # Connect the node to itself for the "continue" action
     cot_node - "continue" >> cot_node