fix tests after change