Amend 3cc0cea

Somehow I lost a part of this commit.

Signed-off-by: Sebastian Wicki <[email protected]>
Signed-off-by: Michi Mutsuzaki <[email protected]>

Author: gandro

pkg/server/local_observer.go

@@ -240,6 +240,7 @@ func (s *LocalObserverServer) GetFlows(
 }
 
 func getFlows(
+	ctx context.Context,
 	req *observer.GetFlowsRequest,
 	server observer.Observer_GetFlowsServer,
 	obs GRPCServer,
@@ -273,7 +274,7 @@ func getFlows(
 	}
 
 	for ; ; i++ {
-		flow, err := flowsReader.Next(server.Context())
+		flow, err := flowsReader.Next(ctx)
 		if err != nil {
 			if err == io.EOF {
 				return nil