Fix pypi installation