diff --git a/action_plugins/qubesformation.py b/action_plugins/qubesformation.py
index 1da1824..db87876 100644
--- a/action_plugins/qubesformation.py
+++ b/action_plugins/qubesformation.py
@@ -22,15 +22,15 @@ def generate_datastructure(vms, task_vars):
         # This allows you to deploy in multiple stages which is useful in cases
         # where you want to create a template after another template is already provisioned.
         print(task_vars)
-        print(task_vars.formation_vm_groups)
-        if hasattr(task_vars, 'formation_vm_groups'):
+        print(task_vars.ansible_facts.formation_vm_groups)
+        if hasattr(task_vars.ansible_facts, 'formation_vm_groups'):
             continueLoop = true
             print("formation")
-            print(task_vars['formation_vm_groups'])
-            for group in task_vars['formation_vm_groups']:
+            print(task_vars.ansible_facts['formation_vm_groups'])
+            for group in task_vars.ansible_facts['formation_vm_groups']:
                 print("group vars")
-                print(task_vars.groups[group])
-                if n in task_vars.groups[group]:
+                print(task_vars.ansible_facts.groups[group])
+                if n in task_vars.ansible_facts.groups[group]:
                     continueLoop = false
             if continueLoop:
                 continue