diff --git a/tests/test_azure_generator.py b/tests/test_azure_generator.py
index da3c2e93..4da2b9d5 100644
--- a/tests/test_azure_generator.py
+++ b/tests/test_azure_generator.py
@@ -222,7 +222,7 @@ def setUp(self):
             "usage_quantity": self.usage_quantity,
             "resource_rate": self.resource_rate,
             "pre_tax_cost": self.pre_tax_cost,
-            "version_two": True,
+            "resource_group": True,
         }
         self.two_hours_ago = (self.now - self.one_hour) - self.one_hour