feat: update tests