Fix error with menu