From 9ef5af3894fec6ff7ddf525e3f619ab80b877542 Mon Sep 17 00:00:00 2001
From: Evelyn <evelyn@e-s.me.uk>
Date: Tue, 30 May 2017 22:25:12 +0100
Subject: [PATCH] NR: Prefer LDBSVWS for headcodes

---
 modules/nr.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/modules/nr.py b/modules/nr.py
index 4b2f5485..e816bbe5 100644
--- a/modules/nr.py
+++ b/modules/nr.py
@@ -278,9 +278,8 @@ class Module(object):
             if query:
                 query = client.service.GetServiceDetailsByRID(rid)
             if schedule:
-                if not query: query = {}
+                if not query: query = {"trainid": schedule["schedule_segment"]["signalling_id"]}
                 for k,v in {
-                    "trainid": schedule["schedule_segment"]["signalling_id"],
                     "operatorCode": schedule["atoc_code"],
                     "serviceType": "class " + schedule_query["tops_inferred"] if schedule_query["tops_inferred"] else SCHEDULE_STATUS.get(schedule["train_status"], "?"),
                 }.items():