fix the pickle problem for models with psi statistics