It would be better to use `mock.patch