diff --git a/claude_sonnet-3-7-bedrock.py b/claude_sonnet-3-7-bedrock.py
index 7f61b5a..07eb090 100644
--- a/claude_sonnet-3-7-bedrock.py
+++ b/claude_sonnet-3-7-bedrock.py
@@ -178,7 +178,6 @@ class Pipeline:
         streaming_response = self.bedrock_runtime.converse_stream(**payload)
         thinking_block = None
         for chunk in streaming_response["stream"]:
-            print(chunk)
             if "contentBlockDelta" in chunk:
                 delta = chunk["contentBlockDelta"]["delta"]
                 
@@ -189,8 +188,8 @@ class Pipeline:
                     yield '<thinking>\n' 
                   yield delta["reasoningContent"]["text"]
 
-                if "contentBlockStop" in chunk: #and chunk["contentBlockStop"]["contentBlockIndex"] == thinking_block:
-                  yield '</thinking>\n\n' 
+            if "contentBlockStop" in chunk and chunk["contentBlockStop"]["contentBlockIndex"] == thinking_block:
+                yield '</thinking>\n\n' 
 
                 # Handle regular response text
                 if "text" in delta: