fix problem in unit test