fix test for pypy