Little optimize

Little optimize