--- a/src/visualizer/model/pyviz.cc	Sun Oct 02 04:38:54 2016 +0200
+++ a/src/visualizer/model/pyviz.cc	Sat Oct 01 22:34:24 2016 -0700
@@ -134,6 +134,12 @@ 
   Config::Connect ("/NodeList/*/DeviceList/*/$ns3::WifiNetDevice/Mac/MacRx",
                    MakeCallback (&PyViz::TraceNetDevRxWifi, this));
 
+  Config::Connect ("/NodeList/*/DeviceList/*/$ns3::WaveNetDevice/MacEntities/*/MacTx",
+                   MakeCallback (&PyViz::TraceNetDevTxWifi, this));
+
+  Config::Connect ("/NodeList/*/DeviceList/*/$ns3::WaveNetDevice/MacEntities/*/MacRx",
+                   MakeCallback (&PyViz::TraceNetDevRxWifi, this));
+
   Config::Connect ("/NodeList/*/DeviceList/*/$ns3::CsmaNetDevice/MacTx",
                    MakeCallback (&PyViz::TraceNetDevTxCsma, this));
 
--- a/src/visualizer/visualizer/base.py	Sun Oct 02 04:38:54 2016 +0200
+++ a/src/visualizer/visualizer/base.py	Sat Oct 01 22:34:24 2016 -0700
@@ -5,8 +5,8 @@ 
 import ns.internet
 import ns.mesh
 import ns.wimax
-import ns.wimax
 import ns.lte
+import ns.wave
 
 import gobject
 import os.path
@@ -45,6 +45,7 @@ 
     ns.wimax.BaseStationNetDevice: NetDeviceTraits(is_wireless=True),
     ns.lte.LteUeNetDevice: NetDeviceTraits(is_wireless=True),
     ns.lte.LteEnbNetDevice: NetDeviceTraits(is_wireless=True),
+    ns.wave.WaveNetDevice: NetDeviceTraits(is_wireless=True),
 }
 
 def lookup_netdevice_traits(class_type):