Correction for code sample