Fix pickling for pypy