Yes, @qiangxue already pointed that out. Will refactor it.