diff --git a/modules/ui/popup/+hacks.el b/modules/ui/popup/+hacks.el
index b8c4fb0cc..6aa30cded 100644
--- a/modules/ui/popup/+hacks.el
+++ b/modules/ui/popup/+hacks.el
@@ -281,16 +281,17 @@ instead of switch-to-buffer-*."
 
 
 ;; `which-key'
-(setq which-key-popup-type 'custom
-      which-key-custom-popup-max-dimensions-function (lambda (_) (which-key--side-window-max-dimensions))
-      which-key-custom-hide-popup-function #'which-key--hide-buffer-side-window
-      which-key-custom-show-popup-function
-      (lambda (act-popup-dim)
-        (cl-letf (((symbol-function 'display-buffer-in-side-window)
-                   (lambda (buffer alist)
-                     (+popup-display-buffer-stacked-side-window
-                      buffer (append '((vslot . -9999)) alist)))))
-          (which-key--show-buffer-side-window act-popup-dim))))
+(after! which-key
+  (setq which-key-popup-type 'custom
+        which-key-custom-popup-max-dimensions-function (lambda (_) (which-key--side-window-max-dimensions))
+        which-key-custom-hide-popup-function #'which-key--hide-buffer-side-window
+        which-key-custom-show-popup-function
+        (lambda (act-popup-dim)
+          (cl-letf (((symbol-function 'display-buffer-in-side-window)
+                     (lambda (buffer alist)
+                       (+popup-display-buffer-stacked-side-window
+                        buffer (append '((vslot . -9999)) alist)))))
+            (which-key--show-buffer-side-window act-popup-dim)))))
 
 
 ;; `windmove'