diff --git a/utils/upgrade b/utils/upgrade
index 1beeafb..f0e7d5a 100755
--- a/utils/upgrade
+++ b/utils/upgrade
@@ -385,11 +385,15 @@ def start(args):
 
     if not isinstance(result, Success):
         print(f"Upgrade failed. Reason: {result}")
+        exit_code = 1
     else:
         print("Upgrade completed successfully!")
+        exit_code = 0
 
     print(f"Full upgrade log: {LOG_FILE}")
 
+    return exit_code
+
 
 def main():
     Path(LOG_FILE).parent.mkdir(mode=0o700, parents=True, exist_ok=True)
@@ -412,10 +416,11 @@ def main():
 
     if len(sys.argv[1:]) == 0:
         parser.print_help()
+        return 1
     else:
         args = parser.parse_args(sys.argv[1:])
-        args.func(args)
+        return args.func(args)
 
 
 if __name__ == "__main__":
-    main()
+    sys.exit(main())