better test for more cases