diff --git a/src/devices/wifi/wifi-mode.cc b/src/devices/wifi/wifi-mode.cc
index f2141e31e..ef4f5ce44 100644
--- a/src/devices/wifi/wifi-mode.cc
+++ b/src/devices/wifi/wifi-mode.cc
@@ -3,6 +3,11 @@
 
 namespace ns3 {
 
+bool operator == (const WifiMode &a, const WifiMode &b)
+{
+  return a.GetUid () == b.GetUid ();
+}
+
 uint32_t 
 WifiMode::GetBandwidth (void) const
 {
diff --git a/src/devices/wifi/wifi-mode.h b/src/devices/wifi/wifi-mode.h
index 200e9fb5f..94ecd12dd 100644
--- a/src/devices/wifi/wifi-mode.h
+++ b/src/devices/wifi/wifi-mode.h
@@ -70,6 +70,8 @@ private:
   uint32_t m_uid;
 };
 
+bool operator == (const WifiMode &a, const WifiMode &b);
+
 class WifiModeFactory
 {
 public: