fix sth in unit test