Made the GPU version run again.