diff --git a/tripleo_heat_templates/tests/test_environment_generator.py b/tripleo_heat_templates/tests/test_environment_generator.py
index 94d13c7197..058afe034c 100644
--- a/tripleo_heat_templates/tests/test_environment_generator.py
+++ b/tripleo_heat_templates/tests/test_environment_generator.py
@@ -480,11 +480,13 @@ parameter_defaults:
                            fake_template2, fake_nested_output]
             mock_open.side_effect = mock_se
             if not self.exception:
-                environment_generator.generate_environments('ignored.yaml')
+                environment_generator.generate_environments('ignored.yaml',
+                                                            'environments')
             else:
                 self.assertRaises(self.exception,
                                   environment_generator.generate_environments,
-                                  'ignored.yaml')
+                                  'ignored.yaml',
+                                  'environments')
                 return
         expected = environment_generator._FILE_HEADER + self.expected_output
         with open(fake_output_path) as f: