diff --git a/AlgebraicDataflowArchitectureModel/src/code/ast/Block.java b/AlgebraicDataflowArchitectureModel/src/code/ast/Block.java
index 844691d..97ac135 100644
--- a/AlgebraicDataflowArchitectureModel/src/code/ast/Block.java
+++ b/AlgebraicDataflowArchitectureModel/src/code/ast/Block.java
@@ -51,7 +51,7 @@
 				public String toString() {
 					return code;
 				}
-			}.toString());
+			});
 		}
 	}
 
diff --git a/AlgebraicDataflowArchitectureModel/src/tests/ASTTest.java b/AlgebraicDataflowArchitectureModel/src/tests/ASTTest.java
index c29320f..1bb233e 100644
--- a/AlgebraicDataflowArchitectureModel/src/tests/ASTTest.java
+++ b/AlgebraicDataflowArchitectureModel/src/tests/ASTTest.java
@@ -14,7 +14,7 @@
 
         VariableDeclarationStatement varStmt = new VariableDeclarationStatement();
         varStmt.setModifiers(List.of(new PlainStatement("private")));
-        varStmt.setType(new PrimitiveType(new SimpleName("int")));
+        //varStmt.setType(new PrimitiveType(new SimpleType("int")));
         varStmt.setFragments(List.of(
                 new VariableDeclaration(DataConstraintModel.typeInt, "x"),
                 new VariableDeclaration(DataConstraintModel.typeInt, "sum")