diff --git a/test_conformance/math_brute_force/main.cpp b/test_conformance/math_brute_force/main.cpp
index 74dd5c47..947b9450 100644
--- a/test_conformance/math_brute_force/main.cpp
+++ b/test_conformance/math_brute_force/main.cpp
@@ -769,10 +769,11 @@ test_status InitCL(cl_device_id device)
     IsTininessDetectedBeforeRounding();
 
     cl_platform_id platform;
-    int err = clGetPlatformIDs(1, &platform, NULL);
+    int err = clGetDeviceInfo(gDevice, CL_DEVICE_PLATFORM, sizeof(platform),
+                              &platform, NULL);
     if (err)
     {
-        print_error(err, "clGetPlatformIDs failed");
+        print_error(err, "clGetDeviceInfo for CL_DEVICE_PLATFORM failed");
         return TEST_FAIL;
     }