Fix test unit error.