Fix CUDA tests