unit test fixing