Added small unit test.