Fix run --once test.