Updated test to use the new async helpers