fix tests after last commit