Just a small change in the OpenCL code.