fix issue in test cases