Changed all .Wait() in on async calls in tests to await.
- Recently learnt the hard way that async has to go all the wait, using .Wait() risks deadlocks.