Fix comment in the code