Attempt to make it py3 compatible