1 files changed, 6 insertions, 3 deletions

diff --git a/src/mailman_pgp/rules/signature.py b/src/mailman_pgp/rules/signature.py
index 28d05e4..27544a2 100644
--- a/src/mailman_pgp/rules/signature.py
+++ b/src/mailman_pgp/rules/signature.py
@@ -19,11 +19,14 @@
 
 from mailman.core.i18n import _
 from mailman.interfaces.rules import IRule
+from mailman.interfaces.usermanager import IUserManager
 from public import public
+from zope.component import getUtility
 from zope.interface import implementer
 
 from mailman_pgp.database import query
-from mailman_pgp.model.list import EncryptedMailingList
+from mailman_pgp.model.address import PGPAddress
+from mailman_pgp.model.list import PGPMailingList
 from mailman_pgp.pgp.wrapper import PGPWrapper
 
 
@@ -47,10 +50,10 @@ class Signature:
 
     def check(self, mlist, msg, msgdata):
         """See `IRule`."""
-        enc_list = query(EncryptedMailingList).filter_by(
+        enc_list = query(PGPMailingList).filter_by(
             list_id=mlist.list_id).first()
         if enc_list is None:
-            raise ValueError('Encrypted mailing list not found.')
+            raise ValueError('PGP enabled mailing list not found.')
         wrapped = PGPWrapper(msg)
 
         if not wrapped.is_signed():