Fixing code sample