Bug fix for use_gpu