Update for async test