From f793054c89c9fa0feec2206881876c7d489a1352 Mon Sep 17 00:00:00 2001
From: hneemann <helmut.neemann@arcor.de>
Date: Wed, 6 Jan 2021 22:09:26 +0100
Subject: [PATCH] minor cleanup

---
 .../de/neemann/digital/draw/library/ResolveGenerics.java | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/main/java/de/neemann/digital/draw/library/ResolveGenerics.java b/src/main/java/de/neemann/digital/draw/library/ResolveGenerics.java
index 78c5a841c..15f69d82f 100644
--- a/src/main/java/de/neemann/digital/draw/library/ResolveGenerics.java
+++ b/src/main/java/de/neemann/digital/draw/library/ResolveGenerics.java
@@ -132,14 +132,13 @@ public class ResolveGenerics {
 
                     boolean isCustom = elementTypeDescription instanceof ElementTypeDescriptionCustom;
                     Statement genS = getStatement(gen);
-                    Context mod = createContext(c, newComponents);
+                    Context mod = createContext(c, newComponents)
+                            .declareVar("args", args);
                     if (isCustom) {
-                        mod.declareVar("args", args)
-                                .declareFunc("setCircuit", new SetCircuitFunc(ve));
+                        mod.declareFunc("setCircuit", new SetCircuitFunc(ve));
                         genS.execute(mod);
                     } else {
-                        mod.declareVar("args", args)
-                                .declareVar("this", new SubstituteLibrary.AllowSetAttributes(elementAttributes));
+                        mod.declareVar("this", new SubstituteLibrary.AllowSetAttributes(elementAttributes));
                         genS.execute(mod);
                     }
                     elementAttributes.putToCache(GEN_ARGS_KEY, mod);