Fix reviewer comment