Fix pypi package