update tests after class changes