Fixed some tests on PyPy