diff --git a/kmip/pie/client.py b/kmip/pie/client.py index d27ad4b1..dcbb7004 100644 --- a/kmip/pie/client.py +++ b/kmip/pie/client.py @@ -174,7 +174,7 @@ def open(self): self._is_open = True except Exception as e: self.logger.error("could not open client connection: %s", e) - raise + raise exceptions.ClientConnectionFailure def close(self): """ @@ -191,7 +191,7 @@ def close(self): self._is_open = False except Exception as e: self.logger.error("could not close client connection: %s", e) - raise + raise exceptions.ClientConnectionFailure @is_connected def create(self, algorithm, length, operation_policy_name=None, name=None,