bugfix
fix tests