diff --git a/src/index.js b/src/index.js
index 589af79..2b836af 100644
--- a/src/index.js
+++ b/src/index.js
@@ -126,7 +126,7 @@ const ReactBridge = {
               return;
             }
 
-            entity.off();
+            entity.off(null, this.updateComponentState);
           }
 
           render() {