Closed
Description
We had to remove the test in 2f3db22 since it was flaky, so we should fix it and add it back.
@johanbrandhorst I've gone ahead and assigned this to you since you wrote the original code and removed the test, but let us know if you don't have time to fix it and we can assign it to someone on our team.