fix tests after class change