Improved async unit tests