fix a little code review