Fix typing of "WHILE_LWT" statements