From 3a871717dbd82f3ca7fea58164a39c652458a218 Mon Sep 17 00:00:00 2001
From: Ho <fan@scroll.io>
Date: Wed, 21 Jan 2026 22:14:31 +0900
Subject: [PATCH] fix issue

---
 coordinator/internal/controller/proxy/client.go         | 2 +-
 coordinator/internal/controller/proxy/prover_session.go | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/coordinator/internal/controller/proxy/client.go b/coordinator/internal/controller/proxy/client.go
index 1df9aa489..2eaf3f77d 100644
--- a/coordinator/internal/controller/proxy/client.go
+++ b/coordinator/internal/controller/proxy/client.go
@@ -123,8 +123,8 @@ func (c *upClient) Login(ctx context.Context, genLogin func(string) (*types.Logi
 }
 
 func handleHttpResp(resp *http.Response) (*ctypes.Response, error) {
+	defer resp.Body.Close()
 	if resp.StatusCode == http.StatusOK || resp.StatusCode == http.StatusUnauthorized {
-		defer resp.Body.Close()
 		var respWithData ctypes.Response
 		// Note: Body is consumed after decoding, caller should not read it again
 		if err := json.NewDecoder(resp.Body).Decode(&respWithData); err == nil {
diff --git a/coordinator/internal/controller/proxy/prover_session.go b/coordinator/internal/controller/proxy/prover_session.go
index b3e30d0a0..76af0a84b 100644
--- a/coordinator/internal/controller/proxy/prover_session.go
+++ b/coordinator/internal/controller/proxy/prover_session.go
@@ -114,12 +114,13 @@ func (c *proverSession) maintainLogin(ctx context.Context, cliMgr Client, up str
 		waitctx := c.completionCtx
 		c.Unlock()
 		select {
-		case <-waitctx.Done():
-			return c.maintainLogin(ctx, cliMgr, up, param, phase)
 		case <-ctx.Done():
 			nerr = fmt.Errorf("ctx fail")
 			return
+		default:
 		}
+		<-waitctx.Done()
+		return c.maintainLogin(ctx, cliMgr, up, param, phase)
 	}
 
 	if phase < curPhase {