Fix broken test
2 files changed