fix code review comment