1 files changed, 5 insertions, 3 deletions

diff --git a/src/mailman_pgp/plugin.py b/src/mailman_pgp/plugin.py
index 37b4304..95006f8 100644
--- a/src/mailman_pgp/plugin.py
+++ b/src/mailman_pgp/plugin.py
@@ -24,8 +24,8 @@ from zope.event import classhandler
 from zope.interface import implementer
 
 from mailman_pgp.config import config
-from mailman_pgp.database import Database, transaction
-from mailman_pgp.model.list import EncryptedMailingList
+from mailman_pgp.database import Database, query, transaction
+from mailman_pgp.model.list import PGPMailingList
 from mailman_pgp.pgp import PGP
 from mailman_pgp.rest.root import RESTRoot
 
@@ -33,6 +33,8 @@ from mailman_pgp.rest.root import RESTRoot
 @public
 @implementer(IPlugin)
 class PGPMailman:
+    """PGP plugin for Mailman!"""
+
     def pre_hook(self):
         """See `IPlugin`."""
         config.load(self.name)
@@ -50,7 +52,7 @@ class PGPMailman:
 
 @classhandler.handler(ListDeletedEvent)
 def on_delete(mlist):
-    encrypted_list = config.db.session.query(EncryptedMailingList).filter_by(
+    encrypted_list = query(PGPMailingList).filter_by(
         list_id=mlist.list_id).first()
     if encrypted_list:
         with transaction() as session: