Fixing pypy

Code is broken on Pypy