diff --git a/tests/conftest.py b/tests/conftest.py
index a869c55c2..4b1d68e09 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -121,7 +121,7 @@ def _notify_db(notify_api, worker_id):
         yield db
 
         db.session.remove()
-        db.engine.dispose()
+        db.get_engine(notify_api).dispose()
 
 
 @pytest.fixture(scope="function")