1 files changed, 5 insertions, 6 deletions
diff --git a/BaseTools/Source/Python/Ecc/CodeFragmentCollector.py b/BaseTools/Source/Python/Ecc/CodeFragmentCollector.py
index b6fa287cf7..d12232cc6f 100644
--- a/BaseTools/Source/Python/Ecc/CodeFragmentCollector.py
+++ b/BaseTools/Source/Python/Ecc/CodeFragmentCollector.py
@@ -22,7 +22,7 @@ import re
 import Common.LongFilePathOs as os
 import sys
 
-import antlr4
+import antlr3
 from Ecc.CLexer import CLexer
 from Ecc.CParser import CParser
 
@@ -499,14 +499,13 @@ class CodeFragmentCollector:
     def ParseFile(self):
         self.PreprocessFile()
         # restore from ListOfList to ListOfString
-        # print(self.Profile.FileLinesList)
         self.Profile.FileLinesList = ["".join(list) for list in self.Profile.FileLinesList]
         FileStringContents = ''
         for fileLine in self.Profile.FileLinesList:
             FileStringContents += fileLine
-        cStream = antlr4.InputStream(FileStringContents)
+        cStream = antlr3.StringStream(FileStringContents)
         lexer = CLexer(cStream)
-        tStream = antlr4.CommonTokenStream(lexer)
+        tStream = antlr3.CommonTokenStream(lexer)
         parser = CParser(tStream)
         parser.translation_unit()
 
@@ -517,9 +516,9 @@ class CodeFragmentCollector:
         FileStringContents = ''
         for fileLine in self.Profile.FileLinesList:
             FileStringContents += fileLine
-        cStream = antlr4.InputStream(FileStringContents)
+        cStream = antlr3.StringStream(FileStringContents)
         lexer = CLexer(cStream)
-        tStream = antlr4.CommonTokenStream(lexer)
+        tStream = antlr3.CommonTokenStream(lexer)
         parser = CParser(tStream)
         parser.translation_unit()