PyPI package should work now