diff --git a/opencl/test/unit_test/mocks/mock_context.cpp b/opencl/test/unit_test/mocks/mock_context.cpp
index 0d849fffc7..ca5eb00d50 100644
--- a/opencl/test/unit_test/mocks/mock_context.cpp
+++ b/opencl/test/unit_test/mocks/mock_context.cpp
@@ -124,6 +124,8 @@ void MockContext::initializeWithDevices(const ClDeviceVector &devices, bool noSp
             }
         }
     }
+
+    setupContextType();
 }
 
 MockDefaultContext::MockDefaultContext() : MockContext(nullptr, nullptr) {