fix pickle import