Correcting the code example