(fix) update tests