Change tests to async