diff --git a/test/test_qbx.py b/test/test_qbx.py
index a6c2319baffc7ee25c06037d20321a66ac4cfc03..aa86edf876fd9dc9aef4e26c2ce10992cfa73240 100644
--- a/test/test_qbx.py
+++ b/test/test_qbx.py
@@ -26,6 +26,8 @@ import numpy as np
 import sys
 
 import pyopencl as cl
+from pyopencl.tools import (  # noqa
+        pytest_generate_tests_for_pyopencl as pytest_generate_tests)
 
 import logging
 logger = logging.getLogger(__name__)