solving unit test problems