Simplify async_colocated, fixing get_colocation_id_test