fix problem test unit