updated tests to be more async