diff --git a/autogpt_platform/backend/backend/data/__init__.py b/autogpt_platform/backend/backend/data/__init__.py
index 45f8f8ea85..7bd5cdf218 100644
--- a/autogpt_platform/backend/backend/data/__init__.py
+++ b/autogpt_platform/backend/backend/data/__init__.py
@@ -1,14 +1,12 @@
-from .graph import NodeModel
-from .integrations import Webhook  # noqa: F401
-
-# Resolve Webhook forward references
-NodeModel.model_rebuild()
-
-
 def _rebuild_library_agent_preset() -> None:
     """Deferred model rebuild to avoid circular import."""
     from backend.api.features.library.model import LibraryAgentPreset
 
+    from .graph import NodeModel
+    from .integrations import Webhook  # noqa: F401
+
+    # Resolve Webhook forward references
+    NodeModel.model_rebuild()
     LibraryAgentPreset.model_rebuild()