Fixed a comment in the code