Update to comply with React componentDidUpdate API change