diff --git a/.emacs.d/init.el b/.emacs.d/init.el
index bf2ce54..0b60390 100644
--- a/.emacs.d/init.el
+++ b/.emacs.d/init.el
@@ -625,9 +625,9 @@
   :if module-file-suffix)
 
 (use-package whitespace
-  :hook (before-save . whitespace-cleanup)
+  :hook ((before-save . whitespace-cleanup)
+         (prog-mode . whitespace-mode))
   :config
-  (whitespace-mode 1)
   (setq whitespace-line-column nil
         whitespace-style '(face empty tabs lines-tail trailing)))