fixed tests after update