diff --git a/tsp/entry_model.py b/tsp/entry_model.py
index 0e9c8b8..e20cc4f 100644
--- a/tsp/entry_model.py
+++ b/tsp/entry_model.py
@@ -62,8 +62,7 @@ def __init__(self, params, model_type=ModelType.XY_TREE):
         '''
         self.entries = []
         if ENTRIES_KEY in params:
-            entries = params.get(ENTRIES_KEY)
-            for entry in entries:
+            for entry in params.get(ENTRIES_KEY):
                 if model_type == ModelType.TIME_GRAPH_TREE:
                     self.entries.append(TimeGraphEntry(entry))
                 else: