Use channel IDs instead of names (lolsecurity)

1 files changed, 3 insertions, 3 deletions

diff --git a/modules/nr.py b/modules/nr.py
index cbdde909..c127f936 100644
--- a/modules/nr.py
+++ b/modules/nr.py
@@ -69,7 +69,7 @@ class Module(object):
 
         trains = sorted(trains, key=lambda t: int(t["scheduled"].replace(":", "")))
 
-        self.result_map[event["target"].name] = trains
+        self.result_map[event["target"].id] = trains
 
         trains_filtered = []
         train_dest_plat = []
@@ -95,10 +95,10 @@ class Module(object):
         service_id = event["args_split"][0]
 
         if service_id.isdigit():
-            if not event["target"].name in self.result_map:
+            if not event["target"].id in self.result_map:
                 event["stdout"].write("No history")
                 return
-            results = self.result_map[event["target"].name]
+            results = self.result_map[event["target"].id]
             if int(service_id) >= len(results):
                 event["stdout"].write("%s is too high. Remember that the first departure is 0" % service_id)
                 return