diff --git a/tests/app/dao/test_service_email_reply_to_dao.py b/tests/app/dao/test_service_email_reply_to_dao.py index 1db7afae5..6ab956866 100644 --- a/tests/app/dao/test_service_email_reply_to_dao.py +++ b/tests/app/dao/test_service_email_reply_to_dao.py @@ -3,6 +3,7 @@ import uuid import pytest from sqlalchemy.exc import SQLAlchemyError +from app import db from app.dao.service_email_reply_to_dao import ( add_reply_to_email_address_for_service, archive_reply_to_email_address, @@ -186,9 +187,7 @@ def test_update_reply_to_email_address(sample_service): email_address="change_address@email.com", is_default=True, ) - updated_reply_to = ServiceEmailReplyTo.query.get( - ServiceEmailReplyTo, first_reply_to.id - ) + updated_reply_to = db.session.get(ServiceEmailReplyTo, first_reply_to.id) assert updated_reply_to.email_address == "change_address@email.com" assert updated_reply_to.updated_at