Correct the code