minor fix in test cases