modify unit test for return type modification