diff --git a/tests/schema_validation_test.py b/tests/schema_validation_test.py
index c03cced..2460614 100644
--- a/tests/schema_validation_test.py
+++ b/tests/schema_validation_test.py
@@ -1,7 +1,6 @@
 from tempfile import NamedTemporaryFile
 
 import asdf
-import jsonschema
 import pydantic
 import pytest
 import yaml
@@ -117,7 +116,7 @@ def test_validate_fail_on_bad_yaml_file():
         # HACK: It is better that the ASDF's schema validation fails before
         # the pydantic's. However, it seems ASDF deserialize first then do their
         # validation.
-        with pytest.raises((jsonschema.ValidationError, pydantic.ValidationError)):
+        with pytest.raises((asdf.exceptions.ValidationError, pydantic.ValidationError)):
             asdf.open(tempfile.name)