style
fix tests