fix tests, maybe