diff --git a/arbnode/transaction_streamer.go b/arbnode/transaction_streamer.go
index 024a32062f..d845bcd477 100644
--- a/arbnode/transaction_streamer.go
+++ b/arbnode/transaction_streamer.go
@@ -497,7 +497,7 @@ func (s *TransactionStreamer) GetProcessedMessageCount() (arbutil.MessageIndex,
 	if err != nil {
 		return 0, err
 	}
-	digestedHead, err := s.exec.HeadMessageNumber().Await(context.Background())
+	digestedHead, err := s.exec.HeadMessageNumber().Await(s.GetContext())
 	if err != nil {
 		return 0, err
 	}
@@ -1093,7 +1093,11 @@ func (s *TransactionStreamer) ResultAtCount(count arbutil.MessageIndex) (*execut
 	}
 	log.Info(FailedToGetMsgResultFromDB, "count", count)
 
-	msgResult, err := s.exec.ResultAtPos(pos).Await(context.Background())
+	ctx := context.Background()
+	if s.Started() {
+		ctx = s.GetContext()
+	}
+	msgResult, err := s.exec.ResultAtPos(pos).Await(ctx)
 	if err != nil {
 		return nil, err
 	}