Which is Jest way for restoring mocked function

In Sinon's stub it is very easy to restore functionality.

const stub = sinon.stub(fs,"writeFile",()=>{})
...
fs.writeFile.restore()

I am looking to do the same thing with Jest. The closest I get is this ugly code:

const fsWriteFileHolder = fs.writeFile
fs.writeFile = jest.fn()
...
fs.writeFile = fsWriteFileHolder