diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py
index 1fab98da39e..83ece5be862 100644
--- a/invokeai/app/invocations/model.py
+++ b/invokeai/app/invocations/model.py
@@ -3,9 +3,6 @@
 
 from pydantic import BaseModel, ConfigDict, Field
 
-# TODO: Permanent fix for this
-# from invokeai.app.invocations.shared import FreeUConfig
-
 from ...backend.model_management import BaseModelType, ModelType, SubModelType
 from .baseinvocation import (
     BaseInvocation,
@@ -20,6 +17,10 @@
     invocation_output,
 )
 
+# TODO: Permanent fix for this
+# from invokeai.app.invocations.shared import FreeUConfig
+
+
 
 class FreeUConfig(BaseModel):
     """