Fixing up comment