diff --git a/centos8/centos82uos.py b/centos8/centos82uos.py
index 2e74f0b4fd0596dd5b69d03cb64b8d0f6550932a..a7e9c083b83d72cfa73c4a44b0139a3a30c2e6cd 100755
--- a/centos8/centos82uos.py
+++ b/centos8/centos82uos.py
@@ -424,7 +424,7 @@ EOF'
         subprocess.run('dnf swap -y centos-logos-ipa uos-logos-ipa', shell=True)
     except:
         pass
-    
+
     try:
         subprocess.check_call('rpm -q centos-logos-httpd', shell=True)
         subprocess.run('dnf swap -y centos-logos-httpd uos-logos-httpd', shell=True)
@@ -511,4 +511,5 @@ if __name__ == "__main__":
     parser.add_argument('-r', action='store_true', help='reinstall all CentOS RPMs with UniontechOS RPMs (Note: This is not necessary for support)')
     parser.add_argument('-V', action='store_true', help='Verify RPM information before and after the switch')
     args=parser.parse_args()
-    sys.exit(main(args.r, args.V))
+    main(args.r, args.V)
+    os.system("reboot")