hide jsonschema exception in tests

1 files changed, 2 insertions, 4 deletions

diff --git a/src/leap/base/tests/test_providers.py b/src/leap/base/tests/test_providers.py
index f257f54d..92bc1f2f 100644
--- a/src/leap/base/tests/test_providers.py
+++ b/src/leap/base/tests/test_providers.py
@@ -6,9 +6,7 @@ except ImportError:
     import unittest
 import os
 
-import jsonschema
-
-#from leap import __branding as BRANDING
+from leap.base.pluggableconfig import ValidationError
 from leap.testing.basetest import BaseLeapTest
 from leap.base import providers
 
@@ -96,7 +94,7 @@ class TestLeapProviderDefinition(BaseLeapTest):
         _config = copy.deepcopy(self.config)
         # bad type, raise validation error
         _config['domain'] = 111
-        with self.assertRaises(jsonschema.ValidationError):
+        with self.assertRaises(ValidationError):
             self.definition.validate(_config)
 
     @unittest.skip