Fixed tests with pypy