update tests

project-koku · Apr 29, 2024 · d54d1a4 · d54d1a4
1 parent eb6b15d
commit d54d1a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_azure_generator.py b/tests/test_azure_generator.py
@@ -485,14 +485,14 @@ def test_update_data_with_attributes(self):
         directional_attributes_v2 = {"data_direction": "in"}
         directional_attributes_v2.update(self.attributes_v2)
         default_generators = [
-            VNGenerator(
+            DTGenerator(
                 self.two_hours_ago,
                 self.now,
                 self.currency,
                 self.account_info,
                 directional_attributes,
             ),
-            VNGenerator(
+            DTGenerator(
                 self.two_hours_ago,
                 self.now,
                 self.currency,