chore: fix test to reflect recent changes
3 files changed