fix the code