diff --git a/sdk-python/omnilogger/logger.py b/sdk-python/omnilogger/logger.py
index 9581aa38..4048d511 100644
--- a/sdk-python/omnilogger/logger.py
+++ b/sdk-python/omnilogger/logger.py
@@ -48,7 +48,7 @@ def filter(self, record) -> bool:
 
             if "message='API response body'" in msg:
                 try:
-                    self.handle_openai_response(msg)
+                    self.extract_output_from_response(msg)
                 except json.decoder.JSONDecodeError:
                     record.msg = "JSONDecodeError for OpenAI response: " + msg
                     return True