Minor tweak to menu hack