Oops, fix tests