From b84ed9c7befc872a5063c3c0b905db3912bf5ced Mon Sep 17 00:00:00 2001
From: Kenneth Kehl <@kkehl@flexion.us>
Date: Fri, 11 Oct 2024 12:33:52 -0700
Subject: [PATCH] fix core daos

---
 tests/app/dao/notification_dao/test_notification_dao.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/app/dao/notification_dao/test_notification_dao.py b/tests/app/dao/notification_dao/test_notification_dao.py
index 8e81db3a2..e2ac10032 100644
--- a/tests/app/dao/notification_dao/test_notification_dao.py
+++ b/tests/app/dao/notification_dao/test_notification_dao.py
@@ -449,12 +449,12 @@ def test_save_notification_creates_sms(sample_template, sample_job):
 
 def _get_notification_query_all():
     stmt = select(Notification)
-    return db.execute(stmt).scalars().all()
+    return db.session.execute(stmt).scalars().all()
 
 
 def _get_notification_query_one():
     stmt = select(Notification)
-    return db.execute(stmt).scalars().one()
+    return db.session.execute(stmt).scalars().one()
 
 
 def _get_notification_query_count():