diff --git a/src/devices/wifi/wifi-mac-queue.cc b/src/devices/wifi/wifi-mac-queue.cc
index d3355aeab..7a001e5f5 100644
--- a/src/devices/wifi/wifi-mac-queue.cc
+++ b/src/devices/wifi/wifi-mac-queue.cc
@@ -228,7 +228,7 @@ WifiMacQueue::Flush (void)
 }
 
 Mac48Address
-WifiMacQueue::GetAddressForPacket (uint8_t type, PacketQueueI it)
+WifiMacQueue::GetAddressForPacket (enum WifiMacHeader::AddressType type, PacketQueueI it)
 {
   if (type == WifiMacHeader::ADDR1)
     {
diff --git a/src/devices/wifi/wifi-mac-queue.h b/src/devices/wifi/wifi-mac-queue.h
index 584f75ec4..224e04f0d 100644
--- a/src/devices/wifi/wifi-mac-queue.h
+++ b/src/devices/wifi/wifi-mac-queue.h
@@ -105,7 +105,7 @@ private:
   typedef std::list<struct Item>::iterator PacketQueueI;
   
   void Cleanup (void);
-  Mac48Address GetAddressForPacket (uint8_t type, PacketQueueI);
+  Mac48Address GetAddressForPacket (enum WifiMacHeader::AddressType type, PacketQueueI);
   
   struct Item {
     Item (Ptr<const Packet> packet,