Fix #122 [NS3] Implement wireless model scenarios with mobility

[nepi.git] / src / nepi / resources / ns3 / ns3wifiremotestationmanager.py

diff --git a/src/nepi/resources/ns3/ns3wifiremotestationmanager.py b/src/nepi/resources/ns3/ns3wifiremotestationmanager.py
index 49eb52f..764c097 100644 (file)
--- a/src/nepi/resources/ns3/ns3wifiremotestationmanager.py
+++ b/src/nepi/resources/ns3/ns3wifiremotestationmanager.py
@@ -31,8 +31,8 @@ class NS3BaseWifiRemoteStationManager(NS3Base):
 
     @property
     def device(self):
-        from nepi.resources.ns3.ns3device import NS3BaseNetDevice
-        devices = self.get_connected(NS3BaseNetDevice.get_rtype())
+        from nepi.resources.ns3.ns3wifinetdevice import NS3BaseWifiNetDevice
+        devices = self.get_connected(NS3BaseWifiNetDevice.get_rtype())
 
         if not devices: 
             msg = "WifiRemoteStationManager not connected to device"