diff --git a/core/hif/src/ce/ce_main.c b/core/hif/src/ce/ce_main.c
index 216ea979a0..31a64f5cbd 100644
--- a/core/hif/src/ce/ce_main.c
+++ b/core/hif/src/ce/ce_main.c
@@ -755,6 +755,7 @@ hif_pci_ce_recv_data(struct CE_handle *copyeng, void *ce_context,
 		&hif_state->msg_callbacks_current;
 
 	do {
+		hif_pm_runtime_mark_last_busy(scn->hif_sc->dev);
 		cdf_nbuf_unmap_single(scn->cdf_dev,
 				      (cdf_nbuf_t) transfer_context,
 				      CDF_DMA_FROM_DEVICE);
diff --git a/core/hif/src/pcie/if_pci.h b/core/hif/src/pcie/if_pci.h
index 3c275ebf32..806f91aadb 100644
--- a/core/hif/src/pcie/if_pci.h
+++ b/core/hif/src/pcie/if_pci.h
@@ -228,5 +228,7 @@ static inline int hif_pm_runtime_resume(struct device *dev)
 	return cnss_pm_runtime_request(dev, CNSS_PM_RUNTIME_RESUME);
 }
 #endif /* WLAN_OPEN_SOURCE */
+#else
+static inline void hif_pm_runtime_mark_last_busy(struct device *dev) { }
 #endif /* FEATURE_RUNTIME_PM */
 #endif /* __ATH_PCI_H__ */