Fix example breakage