diff --git a/claude_sonnet-3-7-bedrock.py b/claude_sonnet-3-7-bedrock.py
index e976eeb..0b86d29 100644
--- a/claude_sonnet-3-7-bedrock.py
+++ b/claude_sonnet-3-7-bedrock.py
@@ -185,16 +185,16 @@ class Pipeline:
                 
                 # Handle reasoning content (Chain of Thought)
                 if "reasoningContent" in delta and "text" in delta["reasoningContent"]:
-                  #if not thinking_block:
-                    #thinking_block = delta["contentBlockIndex"]
-                   # yield '<thinking>\n'
-                  #print(f"Reasoning content: {delta['reasoningContent']}")
+                  if not thinking_block:
+                    thinking_block = chunk["contentBlockDelta"]["contentBlockIndex"]
+                    yield '<thinking>\n'
+                    print(f"Reasoning content: {thinking_block}")
                   if "contentBlockIndex" in delta:
                     print(delta["contentBlockIndex"]) 
                   yield delta["reasoningContent"]["text"]
 
-                #if "contentBlockStop" in delta and delta["contentBlockStop"]["contentBlockIndex"] == thinking_block:
-                #  yield '</thinking>\n\n'
+                if "contentBlockStop" in delta and delta["contentBlockStop"]["contentBlockIndex"] == thinking_block:
+                  yield '</thinking>\n\n'
 
                 # Handle regular response text
                 if "text" in delta: