Add fix to make the code work with pypy