Requiring mock to be passed in now

Python 3 compatability super-easy super cheesy tweak.