diff --git a/lua/gemini/init.lua b/lua/gemini/init.lua
index 7e8928d..5914ad1 100644
--- a/lua/gemini/init.lua
+++ b/lua/gemini/init.lua
@@ -175,7 +175,10 @@ local function gemini_query(prompt, context)
         for i = #lines, 1, -1 do
             if lines[i]:match("^Assistant: Thinking...") then
                 vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', true)
-                vim.api.nvim_buf_set_lines(chat_bufnr, i, i + 1, false, {"Assistant: " .. response})
+                -- Split response into lines and insert "Assistant: " at the start of first line
+                local response_lines = vim.split(response, "\n")
+                response_lines[1] = "Assistant: " .. response_lines[1]
+                vim.api.nvim_buf_set_lines(chat_bufnr, i, i + 1, false, response_lines)
                 vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', false)
                 break
             end