Fix menu problem