Update for new pystan version