Fix display issue in menu class