fixing tests
fix tests