fix tests :(