make test to cover more cases