1 files changed, 11 insertions, 2 deletions

diff --git a/src/mailman_pgp/workflows/tests/test_mod_approval.py b/src/mailman_pgp/workflows/tests/test_mod_approval.py
index 49e4204..7d57a9b 100644
--- a/src/mailman_pgp/workflows/tests/test_mod_approval.py
+++ b/src/mailman_pgp/workflows/tests/test_mod_approval.py
@@ -33,14 +33,23 @@ from mailman_pgp.pgp.wrapper import PGPWrapper
 from mailman_pgp.testing.layers import PGPConfigLayer
 from mailman_pgp.testing.pgp import load_key
 from mailman_pgp.workflows.key_change import KeyChangeBase
-from mailman_pgp.workflows.mod_approval import ModeratorApprovalMixin
+from mailman_pgp.workflows.mod_approval import (
+    ModeratorKeyChangeApprovalMixin)
 
 
 @implementer(IWorkflow)
-class PGPTestWorkflow(KeyChangeBase, ModeratorApprovalMixin):
+class PGPTestWorkflow(KeyChangeBase, ModeratorKeyChangeApprovalMixin):
     name = 'test-workflow'
     description = ''
     initial_state = 'mod_approval'
+    save_attributes = (
+        'approved',
+    )
+
+    def __init__(self, mlist, pgp_address=None, pubkey=None,
+                 pre_approved=False):
+        KeyChangeBase.__init__(self, mlist, pgp_address, pubkey)
+        ModeratorKeyChangeApprovalMixin.__init__(self, pre_approved)
 
 
 class TestModeratorApprovalMixin(TestCase):