Fix python 2