Wouldn't just using a regular `pretend.stub` be easier here?