fix test unit of the code