Fix menu bug