chore: fix tests
6 files changed