Fix tests on pypy3