Extend unit test a little.