Fix code sample.