Fixes two test cases
2 files changed