automata: Fix dfs

1 files changed, 4 insertions, 4 deletions

diff --git a/lib/automata.py b/lib/automata.py
index a947a61..3386761 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -30,8 +30,8 @@ class Transition:
             arguments = [], param_update_function = None,
             arg_to_param_map = None, set_param = None):
         self.name = name
-        self.origin = orig_state.name
-        self.destination = dest_state.name
+        self.origin = orig_state
+        self.destination = dest_state
         self.energy = energy
         self.energy_function = energy_function
         self.duration = duration
@@ -74,8 +74,8 @@ class Transition:
     def to_json(self):
         ret = {
             'name' : self.name,
-            'origin' : self.origin,
-            'destination' : self.destination,
+            'origin' : self.origin.name,
+            'destination' : self.destination.name,
             'is_interrupt' : self.is_interrupt,
             'arguments' : self.arguments,
             'arg_to_param_map' : self.arg_to_param_map,