diff --git a/index_document_test.go b/index_document_test.go
index f801c21..5279a83 100644
--- a/index_document_test.go
+++ b/index_document_test.go
@@ -1669,6 +1669,7 @@ func TestIndex_UpdateDocumentsByFunction(t *testing.T) {
 	require.True(t, res.EditDocumentsByFunction)
 
 	idx := setupMovieIndex(t, c)
+	t.Cleanup(cleanup(c))
 
 	t.Run("Test Upper Case and Add Sparkles around Movie Titles", func(t *testing.T) {
 		task, err := idx.UpdateDocumentsByFunction(&UpdateDocumentByFunctionRequest{