Fix broken code example