Fixed comment in code