Update test to take async listen in account