fix unpickling error