2 files changed, 4 insertions, 2 deletions

diff --git a/src/mailman_pgp/runners/incoming.py b/src/mailman_pgp/runners/incoming.py
index d59b1a5..1dbf8c9 100644
--- a/src/mailman_pgp/runners/incoming.py
+++ b/src/mailman_pgp/runners/incoming.py
@@ -24,6 +24,7 @@ from mailman.model.mailinglist import MailingList
 from public import public
 
 from mailman_pgp.config import config
+from mailman_pgp.database import query
 from mailman_pgp.model.list import EncryptedMailingList
 from mailman_pgp.pgp.wrapper import PGPWrapper
 
@@ -34,7 +35,7 @@ class IncomingRunner(Runner):
         """See `IRunner`."""
         # Is the message for an encrypted mailing list? If not, pass to default
         # incoming runner. If yes, go on.
-        encrypted_list = config.db.query(EncryptedMailingList).filter_by(
+        encrypted_list = query(EncryptedMailingList).filter_by(
             list_id=mlist.list_id).first()
         if not encrypted_list:
             inq = config.get('queues', 'in')
diff --git a/src/mailman_pgp/runners/outgoing.py b/src/mailman_pgp/runners/outgoing.py
index 3b2f97d..e566769 100644
--- a/src/mailman_pgp/runners/outgoing.py
+++ b/src/mailman_pgp/runners/outgoing.py
@@ -24,6 +24,7 @@ from mailman.model.mailinglist import MailingList
 from public import public
 
 from mailman_pgp.config import config
+from mailman_pgp.database import query
 from mailman_pgp.model.list import EncryptedMailingList
 
 
@@ -31,7 +32,7 @@ from mailman_pgp.model.list import EncryptedMailingList
 class OutgoingRunner(Runner):
     def _dispose(self, mlist: MailingList, msg: Message, msgdata: dict):
         """See `IRunner`."""
-        encrypted_list = config.db.query(EncryptedMailingList).filter_by(
+        encrypted_list = query(EncryptedMailingList).filter_by(
             list_id=mlist.list_id).first()
         if not encrypted_list:
             outq = config.get('queues', 'out')