diff --git a/main.py b/main.py
index 4912d85b44ce1a5e8fb4c81489fc28b3a2de9401..e1bb9de43aa29bf22d0b95a8648aba3793083d22 100755
--- a/main.py
+++ b/main.py
@@ -113,7 +113,7 @@ if __name__ == "__main__" :
     transNames = json.load(open(args.model+"/transitions.json", "r"))[1]
     args.predictedStr = transInfos[0]
     args.predicted = set({colName for colName in args.predictedStr.split(',')})
-    transitionSets = [[Transition(elem) for elem in transNames]for ts in transNames]
+    transitionSets = [[Transition(elem) for elem in ts] for ts in transNames]
     strategy = json.load(open(args.model+"/strategy.json", "r"))
     printTS(transitionSets, sys.stderr)
     Decode.decodeMode(args.debug, args.corpus, args.type, transitionSets, strategy, args.reward, args.predicted, args.model)