diff --git a/lib/controller.js b/lib/controller.js
index ee4f8030d..0bb449868 100644
--- a/lib/controller.js
+++ b/lib/controller.js
@@ -34,7 +34,7 @@ class Controller {
         this.extensions = [
             new ExtensionEntityPublish(this.zigbee, this.mqtt, this.state, this.publishEntityState),
             new ExtensionDeviceReceive(this.zigbee, this.mqtt, this.state, this.publishEntityState),
-            new ExtensionDeviceGroupMembership(this.zigbee, this.mqtt, this.publishEntityState),
+            new ExtensionDeviceGroupMembership(this.zigbee, this.mqtt, this.state, this.publishEntityState),
             new ExtensionDeviceConfigure(this.zigbee, this.mqtt, this.state, this.publishEntityState),
             new ExtensionNetworkMap(this.zigbee, this.mqtt, this.state, this.publishEntityState),
             new ExtensionBridgeConfig(this.zigbee, this.mqtt, this.state, this.publishEntityState),