diff --git a/lib/extension/availability.ts b/lib/extension/availability.ts
index 8f656edc7..c7b569a8c 100644
--- a/lib/extension/availability.ts
+++ b/lib/extension/availability.ts
@@ -243,6 +243,8 @@ export default class Availability extends Extension {
                         options,
                         state,
                         device: device.zh,
+                        /* v8 ignore next */
+                        publish: (payload: KeyValue) => this.publishEntityState(device, payload),
                     };
 
                     try {
diff --git a/lib/extension/publish.ts b/lib/extension/publish.ts
index ac89ba538..fde7ac340 100644
--- a/lib/extension/publish.ts
+++ b/lib/extension/publish.ts
@@ -231,6 +231,8 @@ export default class Publish extends Extension {
                 state: entityState,
                 membersState,
                 mapped: definition,
+                /* v8 ignore next */
+                publish: (payload: KeyValue) => this.publishEntityState(re, payload),
             };
 
             // Strip endpoint name from meta.message properties.