diff --git a/emacs/init.org b/emacs/init.org
index 4c19534..ca21818 100755
--- a/emacs/init.org
+++ b/emacs/init.org
@@ -2692,11 +2692,8 @@ And borrowing some functions from python-pytest, we can get some nosetests suppo
   (general-def nosetests-map "f" #'nosetests-file)
   (general-def nosetests-map "d" #'nosetests-def)
 
-  (add-hook 'python-mode-hook
-	    (lambda ()
-	      (make-local-variable 'which-key-replacement-alist)
-	      (which-key-add-key-based-replacements
-		"C-c t n" "nosetests")))
+  (which-key-add-major-mode-key-based-replacements
+    'python-mode "C-c t n" "nosetests")
 #+END_SRC
 
 * Hy