Make py3 compatible.

Should work in Python 3 now.