diff --git a/init.el b/init.el
index f0a7b17..cb212bf 100644
--- a/init.el
+++ b/init.el
@@ -610,7 +610,6 @@
   (org-babel-no-eval-on-ctrl-c-ctrl-c t)
   (org-confirm-babel-evaluate nil)
   :config
-  (add-to-list 'major-mode-remap-alist '(python-mode . python-ts-mode))
   (let (babel-languages)
     (push '(lua . t) babel-languages)
     (push '(python . t) babel-languages)
@@ -1040,6 +1039,7 @@
                         (require 'treesit-auto)
                         (global-treesit-auto-mode +1)))
   :config
+  (add-to-list 'major-mode-remap-alist '(python-mode . python-ts-mode))
   (treesit-auto-add-to-auto-mode-alist))
 
 (use-package eldoc