diff --git a/tests/test_database.py b/tests/test_database.py
index 52a5daa4c..996d4faa5 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -2,7 +2,7 @@ import json
 import pytest
 
 from uuid import UUID, uuid4
-from sqlalchemy.orm import Session
+from sqlmodel import Session
 
 from fastapi.testclient import TestClient