diff --git a/src/wifi/model/wifi-tx-vector.cc b/src/wifi/model/wifi-tx-vector.cc
index 72845e961..8709ceae4 100644
--- a/src/wifi/model/wifi-tx-vector.cc
+++ b/src/wifi/model/wifi-tx-vector.cc
@@ -529,6 +529,10 @@ WifiTxVector::IsDlOfdma() const
     {
         return false;
     }
+    if (IsEht(m_preamble))
+    {
+        return m_ehtPpduType == 0;
+    }
     if (m_muUserInfos.size() == 1)
     {
         return true;
@@ -552,6 +556,10 @@ WifiTxVector::IsDlMuMimo() const
     {
         return false;
     }
+    if (IsEht(m_preamble))
+    {
+        return m_ehtPpduType == 2;
+    }
     if (m_muUserInfos.size() < 2)
     {
         return false;