diff --git a/pykat/finesse.py b/pykat/finesse.py index fb2ee454739a66241800e692d6a48d1233eaf35d..00ba71568ccf067c6a01993e9f3e86c18ec264d8 100644 --- a/pykat/finesse.py +++ b/pykat/finesse.py @@ -285,10 +285,11 @@ class kat(object): self.phase = int(v[1]) elif(first == "retrace"): v = line.split(" ") - if len(v) != 2: + if len(v) > 2: raise pkex.BasePyKatException("Retrace command `{0}` is incorrect.".format(line)) - else: - self.retrace = int(v[1]) + elif len(v) == 2: + self.retrace = v[1] + elif(first == "deriv_h"): v = line.split(" ") if len(v) != 2: