Fix python support