I'm fixing the code