diff --git a/modules/lang/python/config.el b/modules/lang/python/config.el
index ecea8d652..688bf5bb1 100644
--- a/modules/lang/python/config.el
+++ b/modules/lang/python/config.el
@@ -94,6 +94,17 @@ called.")
         "u" #'anaconda-mode-find-references))
 
 
+(def-package! pyimport
+  :after python
+  :config
+  (map! :map python-mode-map
+        :localleader
+        (:prefix ("i" . "insert")
+          :desc "Missing imports" "m" #'pyimport-insert-missing)
+        (:prefix ("r" . "remove")
+          :desc "Unused imports" "r" #'pyimport-remove-unused)))
+
+
 (def-package! nose
   :commands nose-mode
   :preface (defvar nose-mode-map (make-sparse-keymap))
@@ -206,18 +217,3 @@ called.")
   (add-to-list 'global-mode-string
                '(conda-env-current-name (" conda:" conda-env-current-name " "))
                'append))
-
-
-;; Import managements
-(def-package! pyimport
-  :after python
-  :init
-  (map! :after python
-        :map 'python-mode-map
-        :localleader
-        (:prefix ("i" . "insert")
-          :desc "Missing imports" "m" #'pyimport-insert-missing)
-        (:prefix ("r" . "remove")
-          :desc "Unused imports" "r" #'pyimport-remove-unused)
-        )
-  )