diff --git a/test/test_wrapper.py b/test/test_wrapper.py
index 971ca27d6f959b9adb91843acc5038e35d305505..0b993fe2aa674652b970a4a5debe5cc35fa3d9f7 100644
--- a/test/test_wrapper.py
+++ b/test/test_wrapper.py
@@ -684,7 +684,7 @@ def test_platform_get_devices(platform):
                         getattr(cl.device_type, 'CUSTOM', None)):
             continue
         for dev in devs:
-            assert dev.type == dev_type
+            assert dev.type & dev_type == dev_type
 
 
 def test_user_event(ctx_factory):