Worked on integration tests.

1 files changed, 3 insertions, 3 deletions

diff --git a/service/pixelated/config/services.py b/service/pixelated/config/services.py
index 63208489..cc8da83b 100644
--- a/service/pixelated/config/services.py
+++ b/service/pixelated/config/services.py
@@ -40,7 +40,7 @@ class Services(object):
             pixelated_mailboxes)
 
         self.keymanager = leap_session.nicknym
-        self.draft_service = self.setup_draft_service(pixelated_mailboxes)
+        self.draft_service = self.setup_draft_service(leap_session.mail_store)
 
         yield self.post_setup(soledad_querier, leap_session)
 
@@ -77,5 +77,5 @@ class Services(object):
             soledad_querier,
             search_engine))
 
-    def setup_draft_service(self, pixelated_mailboxes):
-        return DraftService(pixelated_mailboxes)
+    def setup_draft_service(self, mail_store):
+        return DraftService(mail_store)