diff --git a/invokeai/backend/util/test_utils.py b/invokeai/backend/util/test_utils.py
index 09b9de9e98..1c7b538882 100644
--- a/invokeai/backend/util/test_utils.py
+++ b/invokeai/backend/util/test_utils.py
@@ -5,7 +5,7 @@ from typing import Optional, Union
 import pytest
 import torch
 
-from invokeai.app.services.config.config_default import InvokeAIAppConfig
+from invokeai.app.services.config.invokeai_config import InvokeAIAppConfig
 from invokeai.backend.install.model_install_backend import ModelInstall
 from invokeai.backend.model_management.model_manager import ModelInfo
 from invokeai.backend.model_management.models.base import BaseModelType, ModelNotFoundException, ModelType, SubModelType
diff --git a/tests/test_config.py b/tests/test_config.py
index 6d76872a0d..2b2492f6a6 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -192,7 +192,7 @@ def test_deny_nodes(patch_rootdir):
     # must parse config before importing Graph, so its nodes union uses the config
     conf = InvokeAIAppConfig().get_config()
     conf.parse_args(conf=allow_deny_nodes_conf, argv=[])
-    from invokeai.app.services.shared.graph import Graph
+    from invokeai.app.services.graph import Graph
 
     # confirm graph validation fails when using denied node
     Graph(nodes={"1": {"id": "1", "type": "integer"}})