Fix the code slightly.