fix test cases
2 files changed