add mising class_idx to zs_Shrink