Fix test
2 files changed