This is a fix for ENG-10110.

The code here triggered an error because it kept sending messages probably before receiving close message.

This is complicated to reproduce so just carefully reading the code as we are using exactly the same approach as here. I am going to test it more after deploying to staging.