Update to the little optimization