improves unit test