Fixing comment bug