diff --git a/llama2.py b/llama2.py
index ababbde..607be30 100644
--- a/llama2.py
+++ b/llama2.py
@@ -67,7 +67,7 @@ class LlamaCpp(PyOperator):
         return new_messages
     
     def parse_outputs(self, response):
-        return response['choices'][0]['text']
+        return response['choices'][0]['message']['content']
 
     @staticmethod
     def supported_model_names():