Updated unit test.