Updates to the unit tests.