diff --git a/llama_cpp/llama_chat_format.py b/llama_cpp/llama_chat_format.py
index 9a09a28..518acc5 100644
--- a/llama_cpp/llama_chat_format.py
+++ b/llama_cpp/llama_chat_format.py
@@ -32,7 +32,7 @@ def _format_llama2(
     ret = system_message + sep
     for role, message in messages:
         if message:
-            ret += message + " "
+            ret += role + message + " "
         else:
             ret += role + " "
     return ret