fixed a couple of test cases