Fix up example code