more tests

we need more unit test