Corrected the code sample.