fixing the test failure
2 files changed