Use mock on python3 as well as python2