Fix comment issue