diff --git a/packages/guardian-prover-health-check/http/get_health_checks_by_guardian_prover_address_test.go b/packages/guardian-prover-health-check/http/get_health_checks_by_guardian_prover_address_test.go
index 334012099ad..90df51b14d0 100644
--- a/packages/guardian-prover-health-check/http/get_health_checks_by_guardian_prover_address_test.go
+++ b/packages/guardian-prover-health-check/http/get_health_checks_by_guardian_prover_address_test.go
@@ -2,14 +2,15 @@ package http
 
 import (
 	"fmt"
-	"github.com/docker/distribution/context"
 	"net/http"
 	"net/http/httptest"
 	"testing"
 
 	"github.com/cyberhorsey/webutils/testutils"
+	"github.com/docker/distribution/context"
 	"github.com/labstack/echo/v4"
 	"github.com/stretchr/testify/assert"
+
 	guardianproverhealthcheck "github.com/taikoxyz/taiko-mono/packages/guardian-prover-health-check"
 )
 
diff --git a/packages/guardian-prover-health-check/repo/healthcheck_test.go b/packages/guardian-prover-health-check/repo/healthcheck_test.go
index d812df6d519..fff747e3217 100644
--- a/packages/guardian-prover-health-check/repo/healthcheck_test.go
+++ b/packages/guardian-prover-health-check/repo/healthcheck_test.go
@@ -67,7 +67,7 @@ func TestIntegration_HealthCheck_Save(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			err = healthCheckRepo.Save(tt.opts)
+			err = healthCheckRepo.Save(context.Background(), &tt.opts)
 			assert.Equal(t, tt.wantErr, err)
 
 			req, err := http.NewRequest(http.MethodGet, "/healtcheck", nil)
@@ -91,7 +91,7 @@ func TestIntegration_HealthCheck_UptimeByGuardianProverId(t *testing.T) {
 
 	assert.Equal(t, nil, err)
 
-	err = healthCheckRepo.Save(guardianproverhealthcheck.SaveHealthCheckOpts{
+	err = healthCheckRepo.Save(context.Background(), &guardianproverhealthcheck.SaveHealthCheckOpts{
 		GuardianProverID: 1,
 		Alive:            true,
 		ExpectedAddress:  "0x123",
@@ -103,7 +103,7 @@ func TestIntegration_HealthCheck_UptimeByGuardianProverId(t *testing.T) {
 
 	assert.Equal(t, err, nil)
 
-	err = healthCheckRepo.Save(guardianproverhealthcheck.SaveHealthCheckOpts{
+	err = healthCheckRepo.Save(context.Background(), &guardianproverhealthcheck.SaveHealthCheckOpts{
 		GuardianProverID: 1,
 		Alive:            true,
 		ExpectedAddress:  "0x123",