Handle both React 17 and 18

[A1Liu]

ReactDOM.render is no longer fully supported in React 18, but we use it in toolkit.