From 8f7caa2e98fc4e7f36b82544bf51e206513a160c Mon Sep 17 00:00:00 2001
From: jennifersp <44716627+jennifersp@users.noreply.github.com>
Date: Thu, 11 Apr 2024 13:20:13 -0700
Subject: [PATCH] [no-release-notes] fix sysbench test cleanup (#7733)

---
 go/performance/utils/benchmark_runner/sysbench.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/go/performance/utils/benchmark_runner/sysbench.go b/go/performance/utils/benchmark_runner/sysbench.go
index bd2822928d..02e637b492 100644
--- a/go/performance/utils/benchmark_runner/sysbench.go
+++ b/go/performance/utils/benchmark_runner/sysbench.go
@@ -139,6 +139,8 @@ func (t *sysbenchTesterImpl) cleanup(ctx context.Context) error {
 }
 
 func (t *sysbenchTesterImpl) Test(ctx context.Context) (*Result, error) {
+	defer t.cleanup(ctx)
+
 	err := t.prepare(ctx)
 	if err != nil {
 		return nil, err
@@ -151,5 +153,5 @@ func (t *sysbenchTesterImpl) Test(ctx context.Context) (*Result, error) {
 		return nil, err
 	}
 
-	return rs, t.cleanup(ctx)
+	return rs, nil
 }