Issue 371

Removed wait and added assert
I managed to reproduce the fail locally, but I don't really know why the fail occurs.
I removed the failing wait and I added an Assert to the test.
Works great for me.