Temporary fix to compile with CUDA