Fix so that tests pass on pypy.