Fixes flakey test