Small fix to work with PyPy

This works with both Python and PyPy.