Fix mark_non_loop for lambody.