diff --git a/test/test_tosca_generator.py b/test/test_tosca_generator.py
index 0011d31..1d560d2 100644
--- a/test/test_tosca_generator.py
+++ b/test/test_tosca_generator.py
@@ -16,7 +16,7 @@
 
 import unittest
 import os
-from xosgenx.generator import XOSGenerator
+from xosgenx.generator import XOSProcessor
 
 current_dir = os.path.dirname(os.path.realpath(__file__))
 OUTPUT_DIR = os.path.join(current_dir, 'out');
@@ -49,7 +49,7 @@
         args.write_to_file = "single"
         args.dest_file = "basic.yaml"
         args.quiet = False
-        output = XOSGenerator.generate(args)
+        output = XOSProcessor.process(args)
         self.assertIn("name:", output)
         self.assertIn("files:", output)
 
@@ -77,9 +77,9 @@
         args.output = OUTPUT_DIR
         args.write_to_file = 'target'
         args.quiet = False
-        output = XOSGenerator.generate(args)
+        output = XOSProcessor.process(args)
         self.assertEqual(output.count("name:"), 4)
         self.assertIn("prop:", output)
 
 if __name__ == '__main__':
-  unittest.main()
\ No newline at end of file
+  unittest.main()