fix tests?