Python 3 compatibility
Minor bugfix