Corrected one of the code blocks