Fixing code sample.