add: better unit test