diff --git a/init.el b/init.el
index 36479fc..877180e 100644
--- a/init.el
+++ b/init.el
@@ -2338,28 +2338,29 @@ Note that this erases the buffer before doing anything."
         retval)))
   (advice-add 'jupyter-kernel-language-mode-properties
               :filter-return #'my/-jupyter-dont-use-ts-modes)
+
+  (with-eval-after-load 'jupyter-org-client
+    ;; fix some bugs
+    (defun my/-fix-jupyter-org--set-src-block-cache (oldfun)
+      (unless jupyter-org--src-block-cache
+        (setq jupyter-org--src-block-cache
+              (list 'invalid nil (make-marker)
+                    (let ((end (make-marker)))
+                      ;; Move the end marker when text is inserted
+                      (set-marker-insertion-type end t)
+                      end))))
+      (funcall oldfun))
+    (advice-add 'jupyter-org--set-src-block-cache :around
+                #'my/-fix-jupyter-org--set-src-block-cache)
+
+    ;; prevent warning in `execute-extended-command'
+    (defun my/-fix-jupyter-org--define-key-filter (oldfun &rest args)
+      "Prevent calling `jupyter-org--define-key-filter' outside of `org-mode'."
+      (when (derived-mode-p 'org-mode)
+        (apply oldfun args)))
+    (advice-add 'jupyter-org--define-key-filter :around
+                #'my/-fix-jupyter-org--define-key-filter))
   :config
-  ;; fix some bugs
-  (defun my/-fix-jupyter-org--set-src-block-cache (oldfun)
-    (unless jupyter-org--src-block-cache
-      (setq jupyter-org--src-block-cache
-            (list 'invalid nil (make-marker)
-                  (let ((end (make-marker)))
-                    ;; Move the end marker when text is inserted
-                    (set-marker-insertion-type end t)
-                    end))))
-    (funcall oldfun))
-  (advice-add 'jupyter-org--set-src-block-cache :around
-              #'my/-fix-jupyter-org--set-src-block-cache)
-
-  ;; prevent warning in `execute-extended-command'
-  (defun my/-fix-jupyter-org--define-key-filter (oldfun &rest args)
-    "Prevent calling `jupyter-org--define-key-filter' outside of `org-mode'."
-    (when (derived-mode-p 'org-mode)
-      (apply oldfun args)))
-  (advice-add 'jupyter-org--define-key-filter :around
-              #'my/-fix-jupyter-org--define-key-filter)
-
   (face-spec-set 'jupyter-repl-traceback
                  '((default . (:background unspecified)))
                  'face-override-spec)