From e196349b2f77902b322141d3b1eec300682ca4a0 Mon Sep 17 00:00:00 2001
From: Nick Chan <towinchenmi@gmail.com>
Date: Wed, 25 Jan 2023 21:32:45 +0800
Subject: [PATCH] fixes

---
 src/main.c | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/main.c b/src/main.c
index 674f267..c14cc86 100644
--- a/src/main.c
+++ b/src/main.c
@@ -95,10 +95,10 @@ int main(int argc, char *argv[]) {
 	pthread_create(&dfuhelper_thread, NULL, dfuhelper, NULL);
 	pthread_join(dfuhelper_thread, NULL);
 	if (dfuhelper_only)
-		return 0;
+		goto normal_exit;
 	exec_checkra1n();
-	if (pongo_exit)
-		return 0;
+	if (pongo_exit || demote)
+		goto normal_exit;
 pongo:
 	spin = true;
 	if (do_pongo_sleep)
@@ -110,6 +110,7 @@ pongo:
 	{
 		sleep(1);
 	}
+normal_exit:
 	if (access("/usr/bin/curl", F_OK) == 0 && ohio) {
 		LOG(LOG_VERBOSE4, "Ohio");
 		char* ohio_argv[] = {