enhance unit test