diff --git a/src/wifi/model/yans-error-rate-model.cc b/src/wifi/model/yans-error-rate-model.cc
index f8505f6ef..f06c5b375 100644
--- a/src/wifi/model/yans-error-rate-model.cc
+++ b/src/wifi/model/yans-error-rate-model.cc
@@ -19,10 +19,11 @@
  *          Sébastien Deronne <sebastien.deronne@gmail.com>
  */
 
+#include <cmath>
 #include "ns3/log.h"
 #include "yans-error-rate-model.h"
 #include "wifi-utils.h"
-#include "wifi-phy.h"
+#include "wifi-tx-vector.h"
 
 namespace ns3 {