Turns out this is the only change needed for Python2 + 3 compatibility