diff --git a/lua/avante/sidebar.lua b/lua/avante/sidebar.lua
index 734e719..add1e86 100644
--- a/lua/avante/sidebar.lua
+++ b/lua/avante/sidebar.lua
@@ -110,6 +110,8 @@ function Sidebar:close()
   if self.code ~= nil and api.nvim_win_is_valid(self.code.winid) then
     fn.win_gotoid(self.code.winid)
   end
+
+  vim.cmd("wincmd =")
 end
 
 ---@return boolean