diff --git a/backend/alembic/env.py b/backend/alembic/env.py
index f3a1014d..7f9503b4 100644
--- a/backend/alembic/env.py
+++ b/backend/alembic/env.py
@@ -25,6 +25,10 @@ from app.models.ai_session_step import AISessionStep  # noqa: F401
 from app.models.psa_post_log import PsaPostLog  # noqa: F401
 from app.models.psa_member_mapping import PsaMemberMapping  # noqa: F401
 from app.models.script_builder_session import ScriptBuilderSession, ScriptBuilderMessage  # noqa: F401
+from app.models.session_branch import SessionBranch  # noqa: F401
+from app.models.fork_point import ForkPoint  # noqa: F401
+from app.models.session_handoff import SessionHandoff  # noqa: F401
+from app.models.session_resolution_output import SessionResolutionOutput  # noqa: F401
 from app.core.config import settings
 
 # this is the Alembic Config object