diff --git a/src/ReactBlessedComponent.js b/src/ReactBlessedComponent.js
index d705716..914afaa 100644
--- a/src/ReactBlessedComponent.js
+++ b/src/ReactBlessedComponent.js
@@ -132,6 +132,10 @@ export default class ReactBlessedComponent {
           node = ReactBlessedIDOperations.get(this._rootNodeID);
 
     update(node, solveClass(options));
+    
+    if (this._currentElement !== nextElement) {
+      this._currentElement = nextElement
+    }
 
     // Updating children
     const childrenToUse = children === null ? [] : [].concat(children);