diff --git a/migrations/script.py.mako b/migrations/script.py.mako
index c1aad568..12efbe18 100644
--- a/migrations/script.py.mako
+++ b/migrations/script.py.mako
@@ -5,8 +5,9 @@ Revision ID: ${up_revision}
 Revises: ${down_revision | comma,n}
 Create Date: ${create_date}
 """
-from alembic import op
 import sqlalchemy as sa
+import sqlmodel as sqlm
+from alembic import op
 ${imports if imports else ""}
 
 # revision identifiers, used by Alembic.