diff --git a/claude_sonnet-3-7-bedrock.py b/claude_sonnet-3-7-bedrock.py
index 13f70d5..d461c31 100644
--- a/claude_sonnet-3-7-bedrock.py
+++ b/claude_sonnet-3-7-bedrock.py
@@ -176,7 +176,6 @@ class Pipeline:
     def stream_response(self, model_id: str, payload: dict) -> Generator:
         # https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-runtime/client/converse_stream.html
         streaming_response = self.bedrock_runtime.converse_stream(**payload)
-        print(f"{streaming_response} for {payload}")
         thinking_block = None
         for chunk in streaming_response["stream"]:
             if "contentBlockDelta" in chunk:
@@ -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: