update tests

I don't think you need this.