Added support for mock.patch