Fixed a flakey test.