got unit test to work