diff --git a/script/codegen/main.go b/script/codegen/main.go
index db38f1b..56cff82 100644
--- a/script/codegen/main.go
+++ b/script/codegen/main.go
@@ -274,12 +274,11 @@ func main() {
 		if err != nil {
 			panic(err)
 		}
+		defer file.Close()
 		err = tt.Execute(file, v)
 		if err != nil {
 			panic(err)
 		}
-		file.Close()
 		fmt.Println("Generated " + dest + " successfully")
 	}
-
 }