Fix for GPU.