Tweak test_batch_done() a bit to see if it stops failing weirdly on Travis.