fix tests
ad new test