diff --git a/claude_sonnet-3-7-bedrock.py b/claude_sonnet-3-7-bedrock.py
index 07eb090..aa38517 100644
--- a/claude_sonnet-3-7-bedrock.py
+++ b/claude_sonnet-3-7-bedrock.py
@@ -188,7 +188,8 @@ class Pipeline:
                     yield '<thinking>\n' 
                   yield delta["reasoningContent"]["text"]
 
-            if "contentBlockStop" in chunk and chunk["contentBlockStop"]["contentBlockIndex"] == thinking_block:
+            if "contentBlockStop" in chunk: # and chunk["contentBlockStop"]["contentBlockIndex"] == thinking_block:
+                print("Thinking End")
                 yield '</thinking>\n\n' 
 
                 # Handle regular response text