fix fail tests after refactoring test cases