Hacky fix for CUDA init