Fix broken example code