Small fix to menu behavior