Polished off unit test.