diff --git a/tests/kfdtest/src/Dispatch.cpp b/tests/kfdtest/src/Dispatch.cpp
index 6209d8cb59..b35484d733 100644
--- a/tests/kfdtest/src/Dispatch.cpp
+++ b/tests/kfdtest/src/Dispatch.cpp
@@ -74,6 +74,10 @@ void Dispatch::SetSpiPriority(unsigned int priority) {
     m_SpiPriority = priority;
 }
 
+void Dispatch::SetPriv(bool priv) {
+    m_NeedCwsrWA = priv;
+}
+
 void Dispatch::Submit(BaseQueue& queue) {
     ASSERT_NE(m_pEop, (void*)0);
     EXPECT_EQ(m_FamilyId, queue.GetFamilyId());
diff --git a/tests/kfdtest/src/Dispatch.hpp b/tests/kfdtest/src/Dispatch.hpp
index b43becf01b..8c7b1964e5 100644
--- a/tests/kfdtest/src/Dispatch.hpp
+++ b/tests/kfdtest/src/Dispatch.hpp
@@ -45,6 +45,8 @@ class Dispatch {
     void SetScratch(int numWaves, int waveSize, HSAuint64 scratch_base);
 
     void SetSpiPriority(unsigned int priority);
+    
+    void SetPriv(bool priv);
 
     HsaEvent *GetHsaEvent() { return m_pEop; }