remove warnings
diff --git a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/SketchToolBehaviour.java b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/SketchToolBehaviour.java
index a3e7d17..a17bd52 100644
--- a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/SketchToolBehaviour.java
+++ b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/SketchToolBehaviour.java
@@ -113,7 +113,7 @@
 
 	private static final double[] ZOOM_LEVELS = new double[] { 0.05, 0.1, 0.5, 1, 1.5, 2, 3, 4 };
 
-	private static final boolean TEST_SHOW_WARNING_DECORATORS = false;
+	private static boolean TEST_SHOW_WARNING_DECORATORS = false;
 
 	private final ISelectionInfo selectionInfo = new SelectionInfoImpl(IColorConstant.BLUE, IColorConstant.LIGHT_BLUE, IColorConstant.RED,
 			LineStyle.DOT);
diff --git a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/AddDiagramFeature.java b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/AddDiagramFeature.java
index e5275f7..4d84276 100644
--- a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/AddDiagramFeature.java
+++ b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/AddDiagramFeature.java
@@ -15,13 +15,12 @@
  *******************************************************************************/
 package org.eclipse.graphiti.testtool.sketch.features;
 
-import java.util.Collection;
-
 import org.eclipse.graphiti.features.IFeatureProvider;
 import org.eclipse.graphiti.features.context.IAddContext;
 import org.eclipse.graphiti.features.impl.AbstractAddShapeFeature;
 import org.eclipse.graphiti.mm.pictograms.Diagram;
 import org.eclipse.graphiti.mm.pictograms.PictogramElement;
+import org.eclipse.graphiti.mm.pictograms.Shape;
 
 /**
  * The Class AddDiagramFeature.
@@ -39,7 +38,10 @@
 	}
 
 	public PictogramElement add(IAddContext context) {
-		((Collection) context.getTargetContainer().getChildren()).add(context.getNewObject());
+		if (context.getNewObject() instanceof Shape) {
+			Shape shape = (Shape) context.getNewObject();
+			context.getTargetContainer().getChildren().add(shape);
+		}
 		return null;
 	}
 
diff --git a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/create/AbstractSketchCreateSimpleConnectionFeature.java b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/create/AbstractSketchCreateSimpleConnectionFeature.java
index 9523819..de2b315 100644
--- a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/create/AbstractSketchCreateSimpleConnectionFeature.java
+++ b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/create/AbstractSketchCreateSimpleConnectionFeature.java
@@ -47,6 +47,7 @@
 		super(fp, name, description);
 	}
 
+	@SuppressWarnings("unused")
 	public Connection create(ICreateConnectionContext context) {
 
 		Anchor startAnchor = context.getSourceAnchor();
diff --git a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/pool/LayoutPoolFeature.java b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/pool/LayoutPoolFeature.java
index 479be80..485e36f 100644
--- a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/pool/LayoutPoolFeature.java
+++ b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/pool/LayoutPoolFeature.java
@@ -23,6 +23,7 @@
 import org.eclipse.graphiti.mm.algorithms.GraphicsAlgorithm;
 import org.eclipse.graphiti.mm.pictograms.ContainerShape;
 import org.eclipse.graphiti.mm.pictograms.PictogramElement;
+import org.eclipse.graphiti.mm.pictograms.Shape;
 import org.eclipse.graphiti.services.Graphiti;
 import org.eclipse.graphiti.testtool.sketch.SketchUtil;
 
@@ -53,7 +54,7 @@
 		PictogramElement pe = context.getPictogramElement();
 		if (pe instanceof ContainerShape) {
 			ContainerShape poolCs = (ContainerShape) pe;
-			List lanePes = poolCs.getChildren();
+			List<Shape> lanePes = poolCs.getChildren();
 			int laneCount = lanePes.size();
 			GraphicsAlgorithm poolGa = poolCs.getGraphicsAlgorithm();
 			int poolHeight = poolGa.getHeight();