diff --git a/test/gpu/jit.cpp b/test/gpu/jit.cpp
index 3859f5ad365..ac96c7d0964 100644
--- a/test/gpu/jit.cpp
+++ b/test/gpu/jit.cpp
@@ -386,7 +386,7 @@ TEST_CASE(compile_math)
         if(contains({migraphx::shape::bool_type, migraphx::shape::tuple_type}, t))
             continue;
         auto name = migraphx::shape::cpp_type(t);
-        if((t == migraphx::shape::half_type) or (t == migraphx::shape::bf16_type))
+        if(contains({migraphx::shape::half_type, migraphx::shape::bf16_type}, t))
             name.insert(0, "migraphx::");
         data_types.push_back(name);
         // fp8 doesn't have vectorization support yet, therefore skip it for now.