oops, fix tests