Fix broken code sample