updated unit test.