Fix for CUDA device type