Fixed build for CUDA