Minor fix

eclipse-cognicrypt · Feb 9, 2021 · ea97afa · ea97afa
1 parent c69cebf
commit ea97afa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...enerator/src/main/java/de/cognicrypt/codegenerator/generator/CrySLBasedCodeGenerator.java b/...enerator/src/main/java/de/cognicrypt/codegenerator/generator/CrySLBasedCodeGenerator.java
@@ -1335,7 +1335,7 @@ public boolean visit(MethodDeclaration node) {
 					curMethod.addParameter(new SimpleEntry<String, String>(svd.getName().getFullyQualifiedName(), svd.getType().toString()));
 				}
 				curMethod.setNumberOfVariablesInTemplate(curMethod.getDeclaredVariables().size());
-				templateClass.addMethod(curMethod);
+				//templateClass.addMethod(curMethod);
 				return super.visit(node);
 			}
 
@@ -1345,6 +1345,7 @@ public void endVisit(MethodDeclaration node) {
 				curMethod.setRules(new ArrayList<>(rules));
 				postCGVars.entrySet().forEach(e -> curMethod.addPostCGVars(new SimpleEntry<String, String>(e.getKey().getFullyQualifiedName(), e.getValue().getJavaType())));
 				rules.clear();
+				templateClass.addMethod(curMethod);
 				super.visit(node);
 			}