diff --git a/mt76.h b/mt76.h
index d509f84e..03402e31 100644
--- a/mt76.h
+++ b/mt76.h
@@ -14,6 +14,7 @@
 #include <linux/usb.h>
 #include <linux/average.h>
 #include <linux/soc/mediatek/mtk_wed.h>
+#include <net/netlink.h>
 #include <net/mac80211.h>
 #if LINUX_VERSION_CODE < KERNEL_VERSION(6,6,0)
 #include <net/page_pool.h>
@@ -1473,6 +1474,7 @@ static inline void mt76_testmode_reset(struct mt76_phy *phy, bool disable)
 #endif
 }
 
+extern const struct nla_policy mt76_tm_policy[NUM_MT76_TM_ATTRS];
 
 /* internal */
 static inline struct ieee80211_hw *
diff --git a/testmode.h b/testmode.h
index 5e2792d8..1b0bcae8 100644
--- a/testmode.h
+++ b/testmode.h
@@ -193,6 +193,4 @@ enum mt76_testmode_tx_mode {
 	MT76_TM_TX_MODE_MAX = NUM_MT76_TM_TX_MODES - 1,
 };
 
-extern const struct nla_policy mt76_tm_policy[NUM_MT76_TM_ATTRS];
-
 #endif