Use levelSplit more

2 files changed, 7 insertions, 6 deletions

diff --git a/src/main/java/bjc/rgens/parser/elements/CaseElement.java b/src/main/java/bjc/rgens/parser/elements/CaseElement.java
index f950f6d..386e2b4 100755
--- a/src/main/java/bjc/rgens/parser/elements/CaseElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/CaseElement.java
@@ -147,7 +147,8 @@ public abstract class CaseElement {
 
 				CaseElement base = createElement(parts[0]);
 
-				return new MethodCaseElement(base, Arrays.copyOfRange(parts, 1, parts.length));
+				parts = Arrays.copyOfRange(parts, 1, parts.length);
+				return new MethodCaseElement(base, parts);
 			} else {
 				return new RuleCaseElement(rawCase);
 			}
diff --git a/src/main/java/bjc/rgens/parser/elements/MethodCaseElement.java b/src/main/java/bjc/rgens/parser/elements/MethodCaseElement.java
index 49375bc..13f58e9 100644
--- a/src/main/java/bjc/rgens/parser/elements/MethodCaseElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/MethodCaseElement.java
@@ -3,7 +3,7 @@ package bjc.rgens.parser.elements;
 import bjc.rgens.parser.GenerationState;
 import bjc.rgens.parser.GrammarException;
 import bjc.rgens.parser.elements.CaseElement;
-import bjc.rgens.parser.methods.MethodElement;
+import bjc.rgens.parser.elements.methods.MethodElement;
 
 import java.util.List;
 import java.util.LinkedList;
@@ -14,20 +14,20 @@ public class MethodCaseElement extends CaseElement {
 	public List<MethodElement> methods;
 
 	public MethodCaseElement(CaseElement base, String... rawMethods) {
-		this(base, new MethodElement[0]);	
+		this.base = base;
 
 		// @TODO
 		//
 		// Implement this
 	}
 
-	public MethodCaseElement(CaseElement base, MethodElement... rawMethods) {
+	public MethodCaseElement(CaseElement base, List<MethodElement> rawMethods) {
 		this.base = base;
 
 		methods = new LinkedList<>();
 		
-		for(MethodElement elm : rawMethods) {
-			methods.add(rawMethods);
+		for(MethodElement method : rawMethods) {
+			methods.add(method);
 		}
 	}