1 files changed, 3 insertions, 2 deletions

diff --git a/src/mailman_pgp/rest/lists.py b/src/mailman_pgp/rest/lists.py
index 8f4cf8b..18669ad 100644
--- a/src/mailman_pgp/rest/lists.py
+++ b/src/mailman_pgp/rest/lists.py
@@ -22,6 +22,7 @@ from mailman.rest.helpers import (
 from public import public
 
 from mailman_pgp.config import config
+from mailman_pgp.database import query
 from mailman_pgp.model.list import EncryptedMailingList
 
 
@@ -39,7 +40,7 @@ class _EncryptedBase(CollectionMixin):
 
     def _get_collection(self, request):
         """See `CollectionMixin`."""
-        return config.db.session.query(EncryptedMailingList).all()
+        return query(EncryptedMailingList).all()
 
 
 @public
@@ -53,7 +54,7 @@ class AllEncryptedLists(_EncryptedBase):
 @public
 class AnEncryptedList(_EncryptedBase):
     def __init__(self, list_id):
-        self._mlist = config.db.session.query(EncryptedMailingList).filter_by(
+        self._mlist = query(EncryptedMailingList).filter_by(
             list_id=list_id).first()
 
     def on_get(self, request, response):