[Minor] Fix comment