Update integration test to sort of use correct async testing API