From 79b5ec586a42afced2dd55eb5134f83f74ee98bc Mon Sep 17 00:00:00 2001
From: Oliver Smith <osmith@fb.com>
Date: Fri, 9 Jan 2015 09:51:28 -0800
Subject: [PATCH] Fix for --to breakage

---
 tradecalc.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tradecalc.py b/tradecalc.py
index 20dc3143..2da73045 100644
--- a/tradecalc.py
+++ b/tradecalc.py
@@ -599,11 +599,12 @@ def getBestHops(self, routes, restrictTo=None):
 
         restrictStations = set()
         if restrictTo:
-            for place in restrictStations:
+            for place in restrictTo:
                 if isinstance(place, Station):
                     restrictStations.add(place)
                 elif isinstance(place, System) and place.stations:
                     restrictStations += place.stations
+        restrictStations = set(restrictStations)
 
         for route in routes:
             tdenv.DEBUG1("Route = {}", route.str())