Fix for CUDA Build