fix missing pickle import