diff --git a/chat_templates/google/gemma-2-2b-it.jinja b/chat_templates/google/gemma-2-2b-it.jinja
index c29c4e2..23129b0 100644
--- a/chat_templates/google/gemma-2-2b-it.jinja
+++ b/chat_templates/google/gemma-2-2b-it.jinja
@@ -14,7 +14,7 @@
     {%- if message['role'] == 'user' and loop.first and system_message is defined %}
         {{ '<start_of_turn>' + role + '\n' + system_message + '\n\n' + message['content'] | trim + '<end_of_turn>\n' }}
     {%- elif message['role'] == 'assistant' %}
-        {{ '<start_of_turn>' + role + '\n' }}{% generation %}{{ message['content'] | trim }}{% endgeneration %}{{ '<end_of_turn>\n' }}
+        {{ '<start_of_turn>' + role + '\n' }}{% generation %}{{ message['content'] | trim + '<end_of_turn>\n' }}{% endgeneration %}
     {%- else %}
         {{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}
     {%- endif %}