Fixed bug, forgot to import WifiSignal from model.