Fix Pypy tests