unit test failing issue fixed