Fix return issue in upgrade zenrun