Fixing comment issue