diff --git a/NeoBoot/ubi_reader_arm/ubi_extract_files.py b/NeoBoot/ubi_reader_arm/ubi_extract_files.py
index 7a94848..c4e0384 100644
--- a/NeoBoot/ubi_reader_arm/ubi_extract_files.py
+++ b/NeoBoot/ubi_reader_arm/ubi_extract_files.py
@@ -2,7 +2,11 @@
 
 import os
 import sys
-import argparse_neo
+#import argparse_neo
+try:
+    import argparse
+except: 
+    import argparse_neo
 from ubi import ubi, get_peb_size
 from ubifs import ubifs
 from ubi_io import ubi_file, leb_virtual_file
@@ -11,7 +15,11 @@ if __name__ == '__main__':
     os.system('echo "\n[NeoBoot] Zip file unzipped.\nInstallation in progress, please wait ..."')
     description = 'Extract contents of UBI image.'
     usage = 'ubi_extract_files.py [options] filepath'
-    parser = argparse_neo.ArgumentParser(usage=usage, description=description)
+#    parser = argparse_neo.ArgumentParser(usage=usage, description=description)
+    try:
+        parser = argparse.ArgumentParser(usage=usage, description=description)
+    except: 
+        parser = argparse_neo.ArgumentParser(usage=usage, description=description)
     parser.add_argument('-l', '--log-file', dest='logpath', help='Log output to file output/LOGPATH. (default: ubifs_output.log)')
     parser.add_argument('-k', '--keep-permissions', action='store_true', dest='permissions', help='Maintain file permissions, requires running as root. (default: False)')
     parser.add_argument('-q', '--quiet', action='store_true', dest='quiet', help='Suppress warnings and non-fatal errors. (default: False)')