Updated the unit test.