Small correction in code chunk.