Fix tests on PyPy