Fix cleanup

Snowflake-Labs · Dec 8, 2024 · e97df58 · e97df58
1 parent 086e202
commit e97df58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/sdk/testint/functions_integration_test.go b/pkg/sdk/testint/functions_integration_test.go
@@ -173,7 +173,7 @@ func TestInt_Functions(t *testing.T) {
 		err := client.Functions.CreateForJava(ctx, request)
 		require.NoError(t, err)
 		t.Cleanup(testClientHelper().Function.DropFunctionFunc(t, id))
-		t.Cleanup(testClientHelper().Stage.RemoveFromUserStageFunc(t, targetPath))
+		t.Cleanup(testClientHelper().Stage.RemoveFromUserStageFunc(t, jarName))
 
 		function, err := client.Functions.ShowByID(ctx, id)
 		require.NoError(t, err)
@@ -926,7 +926,7 @@ func TestInt_Functions(t *testing.T) {
 		err := client.Functions.CreateForScala(ctx, request)
 		require.NoError(t, err)
 		t.Cleanup(testClientHelper().Function.DropFunctionFunc(t, id))
-		t.Cleanup(testClientHelper().Stage.RemoveFromUserStageFunc(t, targetPath))
+		t.Cleanup(testClientHelper().Stage.RemoveFromUserStageFunc(t, jarName))
 
 		function, err := client.Functions.ShowByID(ctx, id)
 		require.NoError(t, err)