diff --git a/include/wifi.h b/include/wifi.h
index 6751d06..6b0d4fe 100644
--- a/include/wifi.h
+++ b/include/wifi.h
@@ -616,11 +616,14 @@ struct rtw_ieee80211_bar {
 struct rtw_ieee80211_ht_cap {
 	__le16	cap_info;
 	u8	ampdu_params_info;
-	u8	supp_mcs_set[16];
+	union {
+		u8	supp_mcs_set[16];
+		struct ieee80211_mcs_info mcs;
+	};	
 	__le16	extended_ht_cap_info;
-	__le16		tx_BF_cap_info;
-	u8	       antenna_selection_info;
-} __attribute__ ((packed));
+	__le32	tx_BF_cap_info;
+	u8   antenna_selection_info;
+} __packed;
 
 /**
  * struct rtw_ieee80211_ht_cap - HT additional information