Small fix for menu