Correct mock import for Python < 3.3