Fix menu display bug