fix pickle issue