diff --git a/project/{{project_name}}/src/{% if use_sqlalchemy %}migration{% endif %}/{% if use_sqlalchemy %}versions{% endif %}/{% if use_sqlalchemy %}2024_01_07_2001-fe58b071a045_create_user_table.py{% endif %} b/project/{{project_name}}/src/{% if use_sqlalchemy %}migration{% endif %}/{% if use_sqlalchemy %}versions{% endif %}/{% if use_sqlalchemy %}2024_01_07_2001-fe58b071a045_create_user_table.py{% endif %}
index dc009d7..f60498b 100644
--- a/project/{{project_name}}/src/{% if use_sqlalchemy %}migration{% endif %}/{% if use_sqlalchemy %}versions{% endif %}/{% if use_sqlalchemy %}2024_01_07_2001-fe58b071a045_create_user_table.py{% endif %}	
+++ b/project/{{project_name}}/src/{% if use_sqlalchemy %}migration{% endif %}/{% if use_sqlalchemy %}versions{% endif %}/{% if use_sqlalchemy %}2024_01_07_2001-fe58b071a045_create_user_table.py{% endif %}	
@@ -9,6 +9,7 @@ from typing import Sequence, Union
 
 from alembic import op
 import sqlalchemy as sa
+import os
 
 
 # revision identifiers, used by Alembic.
@@ -17,6 +18,7 @@ down_revision: Union[str, None] = None
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None
 
+DB_SCHEMA = os.environ.get("DB_SCHEMA")
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###