Fix shouldComponentUpdate methods