Another typo. Works on GPUs so far, fixing it for CPUs.