fixed/add 2 test cases