diff --git a/alembic/env.py b/alembic/env.py
index ba1cca0..04fa3cc 100644
--- a/alembic/env.py
+++ b/alembic/env.py
@@ -4,6 +4,22 @@ from sqlalchemy import engine_from_config, pool
 from sqlmodel import SQLModel
 
 from alembic import context
+from nyahome.database import (  # noqa
+    AiiModel,
+    AiiModelPublic,
+    AiiProvider,
+    AiiProviderPublic,
+    Chatroom,
+    ChatroomChat,
+    ChatroomChatAccept,
+    ChatroomChatDelete,
+    ChatroomChatEdit,
+    ChatroomPublic,
+    ChatScript,
+    ModelUploadFile,
+    ModelUser,
+    ScriptTemplate,
+)
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.