Minor tweak for add