Add missing mock module in setup_requires