diff --git a/libs/testdiff/testdiff.go b/libs/testdiff/testdiff.go
index b4ec6fe7d..1e1df727a 100644
--- a/libs/testdiff/testdiff.go
+++ b/libs/testdiff/testdiff.go
@@ -3,7 +3,6 @@ package testdiff
 import (
 	"fmt"
 	"strings"
-	"testing"
 
 	"github.com/databricks/cli/internal/testutil"
 	"github.com/hexops/gotextdiff"
@@ -29,7 +28,7 @@ func AssertEqualTexts(t testutil.TestingT, filename1, filename2, expected, out s
 	}
 }
 
-func AssertEqualJQ(t *testing.T, expectedName, outName, expected, out string, ignorePaths []string) {
+func AssertEqualJQ(t testutil.TestingT, expectedName, outName, expected, out string, ignorePaths []string) {
 	patch, err := jsondiff.CompareJSON([]byte(expected), []byte(out))
 	if err != nil {
 		t.Logf("CompareJSON error for %s vs %s: %s (fallback to textual comparison)", outName, expectedName, err)