diff --git a/src/wifi/wscript b/src/wifi/wscript
index 4b98bf66e..02fd5e736 100644
--- a/src/wifi/wscript
+++ b/src/wifi/wscript
@@ -110,7 +110,6 @@ def build(bld):
         'model/minstrel-wifi-manager.h',
         'model/wifi-mac.h',
         'model/regular-wifi-mac.h',
-        'model/wifi-phy.h',
         'model/supported-rates.h',
         'model/error-rate-model.h',
         'model/yans-error-rate-model.h',
@@ -119,6 +118,8 @@ def build(bld):
         'model/wifi-mac-queue.h',
         'model/dca-txop.h',
         'model/wifi-mac-header.h',
+        'model/wifi-mac-trailer.h',
+        'model/wifi-phy-state-helper.h',
         'model/qos-utils.h',
         'model/edca-txop-n.h',
         'model/msdu-aggregator.h',