Switch to the mock from PyPI