diff --git a/lua/chatgpt_nvim/init.lua b/lua/chatgpt_nvim/init.lua
index 846c843..b905ea8 100644
--- a/lua/chatgpt_nvim/init.lua
+++ b/lua/chatgpt_nvim/init.lua
@@ -46,6 +46,12 @@ function M.run_chatgpt_command()
   vim.fn.input("") -- wait for user to press Enter
 
   while true do
+    local another = vim.fn.input("Do you want to paste another file? (y/n): ")
+    if another:lower() ~= "y" then
+      handler.finish()
+      break
+    end
+
     -- Ask user for filepath
     local filepath = ""
     while true do
@@ -77,12 +83,6 @@ function M.run_chatgpt_command()
 
     handler.write_file(filepath, filecontent)
     print("Wrote file: " .. filepath)
-
-    local another = vim.fn.input("Do you want to paste another file? (y/n): ")
-    if another:lower() ~= "y" then
-      handler.finish()
-      break
-    end
   end
 end