extend unit test