Addressing unit test failure