1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/net/wireless/p54/p54common.c b/drivers/net/wireless/p54/p54common.c
index 8d2df5b6ecbf..0907e6f246e6 100644
--- a/drivers/net/wireless/p54/p54common.c
+++ b/drivers/net/wireless/p54/p54common.c
@@ -2127,7 +2127,7 @@ static void p54_bss_info_changed(struct ieee80211_hw *dev,
 }
 
 static int p54_set_key(struct ieee80211_hw *dev, enum set_key_cmd cmd,
-                       const u8 *local_address, const u8 *address,
+                       struct ieee80211_vif *vif, struct ieee80211_sta *sta,
                        struct ieee80211_key_conf *key)
 {
         struct p54_common *priv = dev->priv;
@@ -2191,8 +2191,8 @@ static int p54_set_key(struct ieee80211_hw *dev, enum set_key_cmd cmd,
         rxkey->entry = key->keyidx;
         rxkey->key_id = key->keyidx;
         rxkey->key_type = algo;
-        if (address)
-                memcpy(rxkey->mac, address, ETH_ALEN);
+        if (sta)
+                memcpy(rxkey->mac, sta->addr, ETH_ALEN);
         else
                 memset(rxkey->mac, ~0, ETH_ALEN);
         if (key->alg != ALG_TKIP) {