diff --git a/pkg/kv/postgres/store_test.go b/pkg/kv/postgres/store_test.go
index a4c195dadb6..b4628cfe370 100644
--- a/pkg/kv/postgres/store_test.go
+++ b/pkg/kv/postgres/store_test.go
@@ -25,7 +25,7 @@ func TestPostgresKV(t *testing.T) {
 		if err != nil {
 			t.Fatalf("Unable to connect to database: %v", err)
 		}
-		defer conn.Close(context.Background())
+		defer conn.Close(ctx)
 
 		// create a new schema per test
 		schemaName := "test_schema" + testutil.UniqueName()