diff --git a/Kbuild b/Kbuild
index a9ca683..694431f 100644
--- a/Kbuild
+++ b/Kbuild
@@ -208,6 +208,8 @@
 endif
 endif
 
+CONFIG_POWER_MANAGEMENT_OFFLOAD := y
+
 ifeq ($(CONFIG_ROME_IF), snoc)
 	CONFIG_WLAN_TX_FLOW_CONTROL_V2 := y
 endif
@@ -1161,6 +1163,7 @@
 			-I$(WLAN_ROOT)/$(PMO_DIR)/dispatcher/inc \
 			-I$(WLAN_ROOT)/$(PMO_DIR)/dispatcher/src \
 
+ifeq ($(CONFIG_POWER_MANAGEMENT_OFFLOAD), y)
 PMO_OBJS :=     $(PMO_DIR)/core/src/wlan_pmo_main.o \
 		$(PMO_DIR)/core/src/wlan_pmo_apf.o \
 		$(PMO_DIR)/core/src/wlan_pmo_arp.o \
@@ -1185,6 +1188,7 @@
 		$(PMO_DIR)/dispatcher/src/wlan_pmo_tgt_suspend_resume.o \
 		$(PMO_DIR)/dispatcher/src/wlan_pmo_tgt_hw_filter.o \
 		$(PMO_DIR)/dispatcher/src/wlan_pmo_tgt_pkt_filter.o
+endif
 
 ########## DISA (ENCRYPTION TEST) ##########
 
@@ -2158,6 +2162,8 @@
 
 cppflags-$(WLAN_WARN_ON_ASSERT) += -DWLAN_WARN_ON_ASSERT
 
+ccflags-$(CONFIG_POWER_MANAGEMENT_OFFLOAD) += -DWLAN_POWER_MANAGEMENT_OFFLOAD
+
 cppflags-$(CONFIG_WLAN_LOG_FATAL) += -DWLAN_LOG_FATAL
 cppflags-$(CONFIG_WLAN_LOG_ERROR) += -DWLAN_LOG_ERROR
 cppflags-$(CONFIG_WLAN_LOG_WARN) += -DWLAN_LOG_WARN