Updated unit tests.