diff --git a/dcs/action.py b/dcs/action.py
index 46ea0a25..337096ad 100644
--- a/dcs/action.py
+++ b/dcs/action.py
@@ -1638,7 +1638,7 @@ def __init__(self, groupid, task_index):
     def create_from_dict(cls, d, mission):
         # Not sure if this used to be 0/1 in older missions, if
         # so this needs a conditional.
-        return cls(d["set_ai_task"][1], d["ai_task"][2])
+        return cls(d["set_ai_task"][1], d["set_ai_task"][2])
 
     def dict(self):
         d = super(AITaskSet, self).dict()