diff --git a/src/wifi/model/wifi-phy.cc b/src/wifi/model/wifi-phy.cc
index f76f7cdba..8e5b6dbcc 100644
--- a/src/wifi/model/wifi-phy.cc
+++ b/src/wifi/model/wifi-phy.cc
@@ -1994,8 +1994,9 @@ WifiPhy::GetPhyTrainingSymbolDuration (WifiTxVector txVector)
       return MicroSeconds (4 + (4 * Ndltf));
     case WIFI_PREAMBLE_HE_SU:
     case WIFI_PREAMBLE_HE_MU:
-    case WIFI_PREAMBLE_HE_TB:
       return MicroSeconds (4 + (8 * Ndltf));
+    case WIFI_PREAMBLE_HE_TB:
+      return MicroSeconds (8 + (8 * Ndltf));
     default:
       return MicroSeconds (0);
     }
@@ -2043,6 +2044,7 @@ WifiPhy::GetPhySigA2Duration (WifiPreamble preamble)
     case WIFI_PREAMBLE_HE_SU:
     case WIFI_PREAMBLE_VHT_MU:
     case WIFI_PREAMBLE_HE_MU:
+    case WIFI_PREAMBLE_HE_TB:
       //VHT-SIG-A2 and HE-SIG-A2
       return MicroSeconds (4);
     default: