diff --git a/internal/super/generate_test.go b/internal/super/generate_test.go
index 13e0c8884..8a7a37b12 100644
--- a/internal/super/generate_test.go
+++ b/internal/super/generate_test.go
@@ -79,6 +79,10 @@ func TestGenerateNewContractSkipTests(t *testing.T) {
 
 	generateFlags.SkipTests = true
 
+	t.Cleanup(func() {
+		generateFlags.SkipTests = false
+	})
+
 	// Test contract generation
 	_, err := generateNew([]string{"TestContract"}, "contract", logger, state)
 	assert.NoError(t, err, "Failed to generate contract")