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