--- a/src/tools/visualizer/visualizer/plugins/wifi_intrastructure_link.py	Tue Dec 21 22:49:48 2010 -0800
+++ b/src/tools/visualizer/visualizer/plugins/wifi_intrastructure_link.py	Wed Dec 22 12:23:56 2010 +0000
@@ -83,10 +83,10 @@
                 if not isinstance(dev, ns3.WifiNetDevice):
                     continue
                 wifi_mac = dev.GetMac()
-                if isinstance(wifi_mac, ns3.NqstaWifiMac):
+                if isinstance(wifi_mac, ns3.StaWifiMac):
                     wifi_link = WifiLink(viz.links_group, node, dev)
                     self.stations.append((dev, node, wifi_link))
-                elif isinstance(wifi_mac, ns3.NqapWifiMac):
+                elif isinstance(wifi_mac, ns3.ApWifiMac):
                     bssid = ns3.Mac48Address.ConvertFrom(dev.GetAddress())
                     self.access_points[str(bssid)] = node
         #print "APs: ", self.access_points