the issue with this change is that your mock does not assert anymore the argument it receives, which is something we would need to test