diff --git a/init.el b/init.el
index ec3a6ee..b818026 100644
--- a/init.el
+++ b/init.el
@@ -114,12 +114,17 @@
   (line-number-mode +1)
   (column-number-mode +1)
   (size-indication-mode -1)
-  (which-function-mode +1)
   (doom-modeline-mode +1)
   :config
   (set-face-attribute 'doom-modeline nil :weight 'normal)
   (set-face-attribute 'doom-modeline-buffer-path nil :slant 'normal))
 
+(use-package which-func
+  :init
+  (which-function-mode +1)
+  :custom
+  (which-func-modes '(prog-mode)))
+
 (global-visual-line-mode +1)
 
 (use-package visual-fill-column