Fix python 3 build