diff --git a/docs/scripts/swagger_docs.py b/docs/scripts/swagger_docs.py
index 4e80984..1cf7d21 100644
--- a/docs/scripts/swagger_docs.py
+++ b/docs/scripts/swagger_docs.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import os
-from xosgenx.generator import XOSGenerator
+from xosgenx.generator import XOSProcessor
 
 CWD = OUTPUT_DIR = os.path.abspath(os.path.dirname(os.path.realpath(__file__)))
 SWAGGER_DOCS_DIR = os.path.abspath(CWD + '/../swagger/specs')
@@ -40,7 +40,7 @@
     args.dest_file = "swagger.yaml"
     args.quiet = False
     try:
-        XOSGenerator.generate(args)
+        XOSProcessor.process(args)
     except Exception, e:
         print "ERROR: Couldn't generate swagger specs"
         print e
@@ -69,4 +69,4 @@
 
 
 if __name__ == '__main__':
-    main()
\ No newline at end of file
+    main()