chore: fix tests
3 files changed