From 93f6d0a2fc6bc3bdf516125fc736c592c0e6a693 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Deronne?= <sebastien.deronne@gmail.com>
Date: Sun, 3 Feb 2019 16:31:20 +0100
Subject: [PATCH] wifi: Small update for 11ax in Ideal wifi manager

---
 src/wifi/model/ideal-wifi-manager.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/wifi/model/ideal-wifi-manager.cc b/src/wifi/model/ideal-wifi-manager.cc
index a50cbc909..714d8743c 100644
--- a/src/wifi/model/ideal-wifi-manager.cc
+++ b/src/wifi/model/ideal-wifi-manager.cc
@@ -117,7 +117,7 @@ IdealWifiManager::DoInitialize ()
       NS_LOG_DEBUG ("Initialize, adding mode = " << mode.GetUniqueName ());
       AddSnrThreshold (txVector, GetPhy ()->CalculateSnr (txVector, m_ber));
     }
-  // Add all Ht and Vht MCSes
+  // Add all HT, VHT and HE MCSes
   if (GetHtSupported () || GetVhtSupported () || GetHeSupported ())
     {
       nModes = GetPhy ()->GetNMcs ();
@@ -427,7 +427,7 @@ IdealWifiManager::DoGetDataTxVector (WifiRemoteStation *st)
                     {
                       continue;
                     }
-                  for (uint8_t nss = 1; nss <= GetNumberOfSupportedStreams (station); nss++)
+                  for (uint8_t nss = 1; nss <= std::min (GetMaxNumberOfTransmitStreams (), GetNumberOfSupportedStreams (station)); nss++)
                     {
                       txVector.SetNss (nss);
                       if (!txVector.IsValid ())