diff --git a/internal/generator/generator_test.go b/internal/generator/generator_test.go
index 3184256..9ff7958 100644
--- a/internal/generator/generator_test.go
+++ b/internal/generator/generator_test.go
@@ -517,7 +517,7 @@ func loadInterface(t *testing.T, filesCode map[string]input) []*generator.FileCo
 		},
 	}
 	exported := packagestest.Export(t, packagestest.GOPATH, mods)
-	t.Cleanup(exported.Cleanup)
+	defer exported.Cleanup()
 
 	l := loader.NewLoader(func(cfg *packages.Config, patterns ...string) ([]*packages.Package, error) {
 		exported.Config.Mode = cfg.Mode
diff --git a/internal/loader/loader_test.go b/internal/loader/loader_test.go
index 963f841..300afb2 100644
--- a/internal/loader/loader_test.go
+++ b/internal/loader/loader_test.go
@@ -22,7 +22,7 @@ type Service interface {
 }
 `,
 		}}})
-	t.Cleanup(exported.Cleanup)
+	defer exported.Cleanup()
 
 	l := loader.NewLoader(func(cfg *packages.Config, patterns ...string) ([]*packages.Package, error) {
 		exported.Config.Mode = cfg.Mode
@@ -60,7 +60,7 @@ type NotAnInterface struct {
 }
 `,
 		}}})
-	t.Cleanup(exported.Cleanup)
+	defer exported.Cleanup()
 
 	t.Run("RegEx", func(t *testing.T) {
 		l := loader.NewLoader(func(cfg *packages.Config, patterns ...string) ([]*packages.Package, error) {