From ef83fed32cebd7effc7565a40c6e0302d6762834 Mon Sep 17 00:00:00 2001
From: Shreyas Goenka <shreyas.goenka@databricks.com>
Date: Mon, 19 Feb 2024 11:37:56 +0100
Subject: [PATCH] -

---
 internal/helpers.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/internal/helpers.go b/internal/helpers.go
index 91f969e95d..ca5aa25e4a 100644
--- a/internal/helpers.go
+++ b/internal/helpers.go
@@ -539,7 +539,7 @@ func setupLocalFiler(t *testing.T) (filer.Filer, string) {
 	return f, path.Join(filepath.ToSlash(tmp))
 }
 
-func setupWsfsFiler(t *testing.T) (context.Context, filer.Filer) {
+func setupWsfsFiler(t *testing.T) (filer.Filer, string) {
 	t.Log(GetEnvOrSkipTest(t, "CLOUD_ENV"))
 
 	ctx := context.Background()
@@ -555,7 +555,7 @@ func setupWsfsFiler(t *testing.T) (context.Context, filer.Filer) {
 		t.Skip(aerr.Message)
 	}
 
-	return ctx, f
+	return f, tmpdir
 }
 
 func setupDbfsFiler(t *testing.T) (filer.Filer, string) {