fix tests after update