Changed mock import so other platforms can test with lower versions of python