fix tests again