Fix failing tests; add assert_async_task method