Fix python package