diff --git a/d2go/config/__init__.py b/d2go/config/__init__.py
index 5ea4a896..f5b00163 100644
--- a/d2go/config/__init__.py
+++ b/d2go/config/__init__.py
@@ -9,10 +9,10 @@
     CONFIG_CUSTOM_PARSE_REGISTRY,
     CONFIG_SCALING_METHOD_REGISTRY,
     load_full_config_from_file,
-    reroute_config_path,
     temp_defrost,
     temp_new_allowed,
 )
+from d2go.config.utils import reroute_config_path
 
 
 __all__ = [
diff --git a/d2go/config/config.py b/d2go/config/config.py
index 1175d8e3..39020176 100644
--- a/d2go/config/config.py
+++ b/d2go/config/config.py
@@ -5,8 +5,8 @@
 import copy
 import logging
 from typing import List
+from unittest import mock
 
-import mock
 import yaml
 from d2go.config.utils import reroute_config_path, resolve_default_config
 from detectron2.config import CfgNode as _CfgNode