gpu_small fix