diff --git a/libhsakmt/tests/kfdtest/src/KFDTestUtil.cpp b/libhsakmt/tests/kfdtest/src/KFDTestUtil.cpp
index 15d64546fd..46881040ae 100644
--- a/libhsakmt/tests/kfdtest/src/KFDTestUtil.cpp
+++ b/libhsakmt/tests/kfdtest/src/KFDTestUtil.cpp
@@ -207,7 +207,7 @@ unsigned int FamilyIdFromNode(const HsaNodeProperties *props) {
         break;
     case 9:
         familyId = FAMILY_AI;
-        if (props->EngineId.ui32.Minor == 4)
+        if (props->EngineId.ui32.Minor >= 4)
             familyId = FAMILY_AV;
         else if (props->EngineId.ui32.Stepping == 2)
             familyId = FAMILY_RV;