Small optimization tweak