Quick fix for Python 3

It works, but not necessarily the best fix...