Fix python 3, I think