This is for the button not for the menu so handled differently. Will investigate possibility of moving this out...