a bug fix for psi statistics related model pickle