diff --git a/test/test_algorithm.py b/test/test_algorithm.py
index 7dc78288ad722c112535bd27625de42e0cbf395a..9896f284dc533d663a38560df6e1681ac92ba717 100644
--- a/test/test_algorithm.py
+++ b/test/test_algorithm.py
@@ -836,7 +836,7 @@ if __name__ == "__main__":
     if len(sys.argv) > 1:
         exec(sys.argv[1])
     else:
-        from pytest.cmdline import main
+        from py.test.cmdline import main
         main([__file__])
 
 # vim: filetype=pyopencl:fdm=marker
diff --git a/test/test_array.py b/test/test_array.py
index c34c8752a1179c5b3ed4fbbf8ca556874508bac1..f865af05af857a76467ec88826fb1b5227a8c1f9 100644
--- a/test/test_array.py
+++ b/test/test_array.py
@@ -567,7 +567,7 @@ if __name__ == "__main__":
     if len(sys.argv) > 1:
         exec(sys.argv[1])
     else:
-        from pytest.cmdline import main
+        from py.test.cmdline import main
         main([__file__])
 
 # vim: filetype=pyopencl:fdm=marker
diff --git a/test/test_clmath.py b/test/test_clmath.py
index 4eb8ea46d2ae1551e7e44c4313e5694357e0be36..c2d2549089c25b33da3811497e9f82797f4eb11f 100644
--- a/test/test_clmath.py
+++ b/test/test_clmath.py
@@ -299,5 +299,5 @@ if __name__ == "__main__":
     if len(sys.argv) > 1:
         exec(sys.argv[1])
     else:
-        from pytest.cmdline import main
+        from py.test.cmdline import main
         main([__file__])
diff --git a/test/test_wrapper.py b/test/test_wrapper.py
index 0201b74066916ef49563c7c9326f650ae20e5477..01188b4b18bdb77c755a649b8694e06a52891dee 100644
--- a/test/test_wrapper.py
+++ b/test/test_wrapper.py
@@ -516,5 +516,5 @@ if __name__ == "__main__":
     if len(sys.argv) > 1:
         exec(sys.argv[1])
     else:
-        from pytest.cmdline import main
+        from py.test.cmdline import main
         main([__file__])