forgot a minor tweak