Oops... Python 2 compatibility

Accidentally made it Python-2 incompatible while making it
Python-3 compatible. Now works both ways.