fixing a couple tests that weren't waiting for task completion