diff --git a/tests/test_query.py b/tests/test_query.py
index abca97253..88517b92f 100644
--- a/tests/test_query.py
+++ b/tests/test_query.py
@@ -1,5 +1,6 @@
 from typing import Optional
 
+import pytest
 from sqlmodel import Field, Session, SQLModel, create_engine
 
 
@@ -21,6 +22,7 @@ class Hero(SQLModel, table=True):
         session.refresh(hero_1)
 
     with Session(engine) as session:
-        query_hero = session.query(Hero).first()
+        with pytest.warns(DeprecationWarning):
+            query_hero = session.query(Hero).first()
         assert query_hero
         assert query_hero.name == hero_1.name