diff --git a/tests/test_state.py b/tests/test_state.py
index 87a284cd9265df4630475541a63cf391e2928197..36b786b16b86a19ed59f5efa0a473a02b456780f 100644
--- a/tests/test_state.py
+++ b/tests/test_state.py
@@ -121,9 +121,11 @@ def test_state_query():
     rec.insert()
     assert rec.get_property("State") is None
 
-    assert db.execute_query(
-        "FIND TestRT WITH State = State1",
-        unique=True).id == rec.id
+    result = db.execute_query("FIND TestRT WITH State = State1", unique=True)
+    assert result.id == rec.id
+
+    assert result.get_property("State") is None
+    assert result.state == db.State(model="Model1", name="State1")
 
 
 def test_state_transition():