1 files changed, 4 insertions, 1 deletions
diff --git a/src/leap/bitmask/services/soledad/soledadbootstrapper.py b/src/leap/bitmask/services/soledad/soledadbootstrapper.py
index 9e150226..2419fc0d 100644
--- a/src/leap/bitmask/services/soledad/soledadbootstrapper.py
+++ b/src/leap/bitmask/services/soledad/soledadbootstrapper.py
@@ -35,7 +35,7 @@ from leap.common.check import leap_assert, leap_assert_type
 from leap.common.files import get_mtime
 from leap.keymanager import KeyManager, openpgp
 from leap.keymanager.errors import KeyNotFound
-from leap.soledad import Soledad
+from leap.soledad.client import Soledad
 
 logger = logging.getLogger(__name__)
 
@@ -89,6 +89,9 @@ class SoledadBootstrapper(AbstractBootstrapper):
 
     # retries
 
+    def cancel_bootstrap(self):
+        self._soledad_retries = self.MAX_INIT_RETRIES
+
     def should_retry_initialization(self):
         """
         Returns True if we should retry the initialization.