Improved handling of some ARIMA models