Fix tests for pypy3