Fix PyPi package