fix tests
4 files changed