Switch tests to the 'done' method, since they're async.