Fix for python 3 support - thanks @Linovia