fixing up test cases