diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index c1ba022a..c65ad29c 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,7 +1,7 @@
 ci:
   autoupdate_schedule: quarterly
 
-default_stages: [commit]
+default_stages: [pre-commit]
 
 default_install_hook_types: [pre-commit, commit-msg]
 
@@ -36,7 +36,7 @@ repos:
     rev: v2.3.0
     hooks:
       - id: codespell
-        stages: [commit, commit-msg]
+        stages: [pre-commit, commit-msg]
         args: [--ignore-words-list, "nd,te,ois,dscribe", --check-filenames]
 
   - repo: https://github.com/kynan/nbstripout
diff --git a/crystal_toolkit/components/localenv.py b/crystal_toolkit/components/localenv.py
index 8cc6a6dc..2620d5eb 100644
--- a/crystal_toolkit/components/localenv.py
+++ b/crystal_toolkit/components/localenv.py
@@ -21,7 +21,7 @@
     LightStructureEnvironments,
 )
 from pymatgen.analysis.graphs import MoleculeGraph, StructureGraph
-from pymatgen.analysis.local_env import LocalStructOrderParams, CN_OPT_PARAMS
+from pymatgen.analysis.local_env import CN_OPT_PARAMS, LocalStructOrderParams
 from pymatgen.core import Molecule, Structure
 from pymatgen.ext.matproj import MPRester
 from pymatgen.io.ase import AseAtomsAdaptor