Small fix to work with PyPy