Fix code sample