fix tests

good to review