unit test for more cases