Fixed broken test
2 files changed