fix tests after previous commit