update unit test code to support name change