This one actually should fix it. Also, add mock to requirements-text.txt.