fix test
fix tests