From 221bb9a23a454488430443f838888f355dc20d0d Mon Sep 17 00:00:00 2001
From: "David J. Allen" <davidallendj@gmail.com>
Date: Wed, 4 Dec 2024 15:59:05 -0700
Subject: [PATCH] tests: changed how the test generator is added

---
 tests/generate_test.go | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/generate_test.go b/tests/generate_test.go
index d5e400a..ce42bec 100644
--- a/tests/generate_test.go
+++ b/tests/generate_test.go
@@ -355,9 +355,7 @@ func TestGenerateExampleWithServer(t *testing.T) {
 
 	// create new server, add test generator, and start in background
 	server := server.New(&conf)
-	server.GeneratorParams.Generators = map[string]generator.Generator{
-		"test": &gen,
-	}
+	generator.DefaultGenerators["test"] = &gen
 	go server.Serve()
 
 	// make request to server to generate a file