Maybe fix py3 compatability?