diff --git a/src/index.js b/src/index.js
index f5ea9c4..5095b0d 100644
--- a/src/index.js
+++ b/src/index.js
@@ -43,11 +43,15 @@ export default function (CustomElement, opts) {
     static get displayName() {
       return displayName;
     }
+    getNativeElement() {
+      return this._elementNode;
+    }
     componentDidMount() {
+      this._elementNode = ReactDOM.findDOMNode(this);
       this.componentWillReceiveProps(this.props);
     }
     componentWillReceiveProps(props) {
-      const node = ReactDOM.findDOMNode(this);
+      const node = this.getNativeElement();
       Object.keys(props).forEach(name => {
         if (name === 'children' || name === 'style') {
           return;