diff --git a/lib/tfw/yaml_fsm.py b/lib/tfw/yaml_fsm.py
index bb16b12..b4fd45a 100644
--- a/lib/tfw/yaml_fsm.py
+++ b/lib/tfw/yaml_fsm.py
@@ -31,10 +31,10 @@ class YamlFSM(FSMBase):
         topatch = {'on_enter', 'on_exit', 'prepare', 'before', 'after'}
 
         for array in {'states', 'transitions'}:
-            for i, json_obj in enumerate(self.config[array]):
+            for json_obj in self.config[array]:
                 for attribute, value in json_obj.items():
                     if attribute in topatch:
-                        self.config[array][i][attribute] = partial(self.run, value)
+                        json_obj[attribute] = partial(self.run, value)
 
     @staticmethod
     def run(command, event):