Fix the code