Change teh exception

1 files changed, 3 insertions, 5 deletions

diff --git a/tests/hazmat/primitives/test_block.py b/tests/hazmat/primitives/test_block.py
index 4c756203..ea40127e 100644
--- a/tests/hazmat/primitives/test_block.py
+++ b/tests/hazmat/primitives/test_block.py
@@ -18,9 +18,7 @@ import binascii
 import pytest
 
 from cryptography import utils
-from cryptography.exceptions import (
-    UnsupportedAlgorithm, AlreadyFinalized, IncorrectPadding
-)
+from cryptography.exceptions import UnsupportedAlgorithm, AlreadyFinalized
 from cryptography.hazmat.primitives import interfaces
 from cryptography.hazmat.primitives.ciphers import (
     Cipher, algorithms, modes
@@ -119,10 +117,10 @@ class TestCipherContext(object):
         )
         encryptor = cipher.encryptor()
         encryptor.update(b"1")
-        with pytest.raises(IncorrectPadding):
+        with pytest.raises(ValueError):
             encryptor.finalize()
 
         decryptor = cipher.decryptor()
         decryptor.update(b"1")
-        with pytest.raises(IncorrectPadding):
+        with pytest.raises(ValueError):
             decryptor.finalize()