Fixing pypi package