fix

internal/proxy/client.go

@@ -158,7 +158,7 @@ func (c *Client) sendInitialRequest(stream cloudproxyv1alpha.CloudProxyAPI_Strea
 	if err != nil {
 		return fmt.Errorf("stream.Send: initial request %w", err)
 	}
-	c.lastSeenReceive.Store(time.Now().UnixNano())
+	c.lastSeenSend.Store(time.Now().UnixNano())
 
 	c.log.Info("Stream to castai started successfully")
 
@@ -274,6 +274,7 @@ func (c *Client) receive(ctx context.Context, stream cloudproxyv1alpha.CloudProx
 			return fmt.Errorf("stream.Recv: %w", err)
 		}
 
+		c.lastSeenReceive.Store(time.Now().UnixNano())
 		c.log.Debugf("Handling message from castai")
 		go c.handleMessage(stream.Context(), in, respCh)
 	}
@@ -285,7 +286,6 @@ func (c *Client) handleMessage(ctx context.Context, in *cloudproxyv1alpha.Stream
 		return
 	}
 
-	c.lastSeenReceive.Store(time.Now().UnixNano())
 	c.processConfigurationRequest(in)
 
 	// skip processing http request if keep alive message.