Fix test to make CI happy (didn't like done() being called twice)