fix test unit