Fix test exit