Fix tests on pypy