diff --git a/core/hdd/src/wlan_hdd_twt.c b/core/hdd/src/wlan_hdd_twt.c
index e41f434cb96c..4ec29a5540bb 100644
--- a/core/hdd/src/wlan_hdd_twt.c
+++ b/core/hdd/src/wlan_hdd_twt.c
@@ -2686,15 +2686,15 @@ static int hdd_twt_pause_session(struct hdd_adapter *adapter,
 	params.vdev_id = adapter->vdev_id;
 
 	ret = wlan_cfg80211_nla_parse_nested(tb,
-				      QCA_WLAN_VENDOR_ATTR_TWT_NUDGE_MAX,
-				      twt_param_attr,
-				      qca_wlan_vendor_twt_nudge_dialog_policy);
+					QCA_WLAN_VENDOR_ATTR_TWT_SETUP_MAX,
+					twt_param_attr,
+					qca_wlan_vendor_twt_add_dialog_policy);
 	if (ret) {
 		hdd_debug("command parsing failed");
 		return ret;
 	}
 
-	id = QCA_WLAN_VENDOR_ATTR_TWT_NUDGE_FLOW_ID;
+	id = QCA_WLAN_VENDOR_ATTR_TWT_SETUP_FLOW_ID;
 	if (tb[id]) {
 		params.dialog_id = nla_get_u8(tb[id]);
 	} else {