diff --git a/lua/chatgpt_nvim/config.lua b/lua/chatgpt_nvim/config.lua
index c41b500..ba71ef9 100644
--- a/lua/chatgpt_nvim/config.lua
+++ b/lua/chatgpt_nvim/config.lua
@@ -60,7 +60,7 @@ function M.load()
   if data and ok_yaml then
     local ok, result = pcall(lyaml.load, data)
     if ok and type(result) == "table" then
-      local config = result[1]
+      local config = result
       if type(config) == "table" then
         return {
           initial_prompt = config.initial_prompt or "",
diff --git a/lua/chatgpt_nvim/init.lua b/lua/chatgpt_nvim/init.lua
index 1fe3943..98c2014 100644
--- a/lua/chatgpt_nvim/init.lua
+++ b/lua/chatgpt_nvim/init.lua
@@ -21,7 +21,6 @@ local function parse_response(raw)
     vim.api.nvim_err_writeln("Failed to parse YAML response.")
     return nil
   end
-  data = data[1]
   return data
 end