diff --git a/source/tool/chipsec_main.py b/source/tool/chipsec_main.py
index 0dfb2327..f5f63736 100644
--- a/source/tool/chipsec_main.py
+++ b/source/tool/chipsec_main.py
@@ -656,8 +656,10 @@ class ChipsecMain:
         logger().disable()
         return modules_failed
 
-def main():
-    chipsecMain = ChipsecMain(sys.argv[1:])
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv[1:]
+    chipsecMain = ChipsecMain(argv)
     return chipsecMain.main()
 
 if __name__ == "__main__":
diff --git a/source/tool/chipsec_util.py b/source/tool/chipsec_util.py
index 8976df26..fef4e4df 100644
--- a/source/tool/chipsec_util.py
+++ b/source/tool/chipsec_util.py
@@ -207,10 +207,12 @@ class ChipsecUtil:
                       "################################################################" )
         logger().log( "[CHIPSEC] Version %s" % __version__ )
 
-def main():
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv
     chipsecUtil = ChipsecUtil()
     chipsecUtil.print_banner()
-    return chipsecUtil.main(sys.argv)
+    return chipsecUtil.main(argv)
 
        
 if __name__ == "__main__":