diff --git a/lua/piemenu/view/highlight_group.lua b/lua/piemenu/view/highlight_group.lua
index e97a48f..f647fba 100644
--- a/lua/piemenu/view/highlight_group.lua
+++ b/lua/piemenu/view/highlight_group.lua
@@ -17,7 +17,9 @@ local group = vim.api.nvim_create_augroup("piemenu", {})
 vim.api.nvim_create_autocmd({ "ColorScheme" }, {
   group = group,
   pattern = { "*" },
-  callback = setup_highlight_groups,
+  callback = function()
+    setup_highlight_groups()
+  end,
 })
 
 return setup_highlight_groups()