Fix menu not closing