Fixed the issue causing unit test failure