Small code sample fix