fix tests, hopefully