diff --git a/modules/tools/ein/config.el b/modules/tools/ein/config.el
index 3bc10cd1e..f660f0af8 100644
--- a/modules/tools/ein/config.el
+++ b/modules/tools/ein/config.el
@@ -3,7 +3,7 @@
 ;;
 ;;; Packages
 
-(after! ein
+(after! ein-notebook
   (defun +ein-buffer-p (buf)
     (or (memq buf (ein:notebook-opened-buffers))
         (memq buf (mapcar #'ein:notebooklist-get-buffer (ein:notebooklist-keys)))))