From fe378eaf679243ae64e23d49d1524235611af0e8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Deronne?= <sebastien.deronne@gmail.com>
Date: Sun, 18 May 2025 10:54:35 +0200
Subject: [PATCH] wifi: Fix EHT RU allocation for 2x996 tones

---
 src/wifi/model/eht/eht-ru.cc | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/wifi/model/eht/eht-ru.cc b/src/wifi/model/eht/eht-ru.cc
index b0094d1be..4b74764b0 100644
--- a/src/wifi/model/eht/eht-ru.cc
+++ b/src/wifi/model/eht/eht-ru.cc
@@ -833,10 +833,9 @@ EhtRu::GetEqualizedRuAllocation(RuType ruType, bool isOdd, bool hasUsers)
         return hasUsers ? 64 : 28;
     case RuType::RU_484_TONE:
         return hasUsers ? 72 : 29;
-    case RuType::RU_996_TONE:
-        return hasUsers ? 80 : 30;
     default:
-        return hasUsers ? 88 : 30;
+        const auto ruAlloc = (ruType == RuType::RU_2x996_TONE) ? 88 : 80;
+        return hasUsers ? ruAlloc : 30;
     }
 }