Fix argument handing in `while`.