making the previous test cases better