A minor tweak