Fix comments from Lee.