Fix to to more tests that should wait fir results
(but probably didn't matter since the mock has results right away)