fix 2 fail testcase

Make all tests pass.