Mark more tests as async