Updated the unit tests.