Update the calledWith replace