Made one minor tweak