Small fix so Python3 branch also works on Python 2