diff --git a/pkg/diag/diag_test.go b/pkg/diag/diag_test.go
index 2702f31..bb13ecd 100644
--- a/pkg/diag/diag_test.go
+++ b/pkg/diag/diag_test.go
@@ -127,8 +127,10 @@ func TestCheckBridge(t *testing.T) {
 	})
 	t.Run("pass when both lookups resolve", func(t *testing.T) {
 		d := &Diag{
-			Store:    &fakeStore{classes: []nodev1.RuntimeClass{runtimeClass("kata", "kata")}},
-			NewNaive: func() (bridge.Resolver, error) { return &fakeResolver{handlers: map[string]string{"kata": "kata"}}, nil },
+			Store: &fakeStore{classes: []nodev1.RuntimeClass{runtimeClass("kata", "kata")}},
+			NewNaive: func() (bridge.Resolver, error) {
+				return &fakeResolver{handlers: map[string]string{"kata": "kata"}}, nil
+			},
 		}
 		res := d.CheckBridge([]string{"kata"})
 		if res.Status != StatusPass {
@@ -249,7 +251,9 @@ func TestRunHealthy(t *testing.T) {
 	d := &Diag{
 		MetaServer: &fakeMeta{classes: []nodev1.RuntimeClass{runtimeClass("kata", "kata")}},
 		Store:      &fakeStore{classes: []nodev1.RuntimeClass{runtimeClass("kata", "kata")}},
-		NewNaive:   func() (bridge.Resolver, error) { return &fakeResolver{handlers: map[string]string{"kata": "kata"}}, nil },
+		NewNaive: func() (bridge.Resolver, error) {
+			return &fakeResolver{handlers: map[string]string{"kata": "kata"}}, nil
+		},
 	}
 	d = withPaths(t, d)
 	report := d.Run(context.Background(), nil)