diff --git a/qahw_api/test/qap_wrapper_extn.c b/qahw_api/test/qap_wrapper_extn.c
index 4b27674..a084277 100644
--- a/qahw_api/test/qap_wrapper_extn.c
+++ b/qahw_api/test/qap_wrapper_extn.c
@@ -1719,8 +1719,14 @@
             input_config.flags = QAP_MODULE_FLAG_SYSTEM_SOUND;
             stream_info->system_input = true;
         } else {
-            ALOGV("%s::%d Set Primary Main Input Flag", __func__, __LINE__);
-            input_config.flags = QAP_MODULE_FLAG_PRIMARY;
+            if (stream_info->flags & AUDIO_OUTPUT_FLAG_ASSOCIATED) {
+                ALOGV("%s::%d Set Secondary Assoc Input Flag", __func__, __LINE__);
+                input_config.flags = QAP_MODULE_FLAG_SECONDARY;
+                stream_info->sec_input = true;
+            } else {
+                ALOGV("%s::%d Set Primary Main Input Flag", __func__, __LINE__);
+                input_config.flags = QAP_MODULE_FLAG_PRIMARY;
+            }
         }
     }