can you add a unit test ?