Why is the added waiting necessary in the test now that you've fixed this? That isn't immediately obvious to me.