Fixed the broken build because of the unit test case failure.