diff --git a/test/test_extension.py b/test/test_extension.py
index cf0e815..505ffcf 100644
--- a/test/test_extension.py
+++ b/test/test_extension.py
@@ -1,10 +1,11 @@
-import pytest
 import aenum
+import pytest
 
 import socials_util
 
+
 def test_re_social_path_validation():
-    with pytest.raises(ValueError):
+    with pytest.raises(ValueError, match='Redundant www: www.example.org'):
         socials_util.re_social_path('www.example.org')
 
 
diff --git a/test/test_wikidata_property.py b/test/test_wikidata_property.py
index 909a26a..79303af 100644
--- a/test/test_wikidata_property.py
+++ b/test/test_wikidata_property.py
@@ -1,4 +1,4 @@
-import wikidata
+import wikidata.client
 
 import socials_util