diff --git a/test/test_persistent_dict.py b/test/test_persistent_dict.py
index ea6665f34ebb6efa3dd380171ad3f68574f29900..1318bbb0387d8d9e1e9351ea7902cec2df00b510 100644
--- a/test/test_persistent_dict.py
+++ b/test/test_persistent_dict.py
@@ -318,5 +318,5 @@ if __name__ == "__main__":
     if len(sys.argv) > 1:
         exec(sys.argv[1])
     else:
-        from py.test.cmdline import main
+        from pytest import main
         main([__file__])
diff --git a/test/test_pytools.py b/test/test_pytools.py
index 5346c25591d142a5af4c3d677ec8b92d5f03bbe5..fbd7cf78bb0859268a72894cf53b3d78e986104b 100644
--- a/test/test_pytools.py
+++ b/test/test_pytools.py
@@ -210,5 +210,5 @@ if __name__ == "__main__":
     if len(sys.argv) > 1:
         exec(sys.argv[1])
     else:
-        from py.test.cmdline import main
+        from pytest import main
         main([__file__])