diff --git a/tests/cloudpickle_test.py b/tests/cloudpickle_test.py
index 768817318..68499f3f4 100644
--- a/tests/cloudpickle_test.py
+++ b/tests/cloudpickle_test.py
@@ -714,10 +714,10 @@ def f():
             s = py.builtin.set([1])
             return s.pop()
 
-        # some setup is required to allow pytest apimodules to be correctly serializable.
+        # some setup is required to allow pytest apimodules to be correctly
+        # serializable.
         from cloudpickle import CloudPickler
-        from py._apipkg import ApiModule
-        CloudPickler.dispatch[ApiModule] = CloudPickler.save_module
+        CloudPickler.dispatch[type(py.builtin)] = CloudPickler.save_module
         g = cloudpickle.loads(cloudpickle.dumps(f))
 
         result = g()