Fix test unit