fix tests ....