fix tests
update tests