Update function to fix OG menus