update tests

fix tests

squashme