Fix up for PyPi