CUDA fix