fixing code sample