try to write unit test