chore: update with test