alter unit test code.