Minor fixup
Minor tweak