Fix tests

Should have read:
http://www.voidspace.org.uk/python/mock/patch.html#where-to-patch

:P