fix broken code sample