YH-177: Fix menu class