fix tests, shame on me