diff --git a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/SketchTextProposalDirectEditingFeature.java b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/SketchTextProposalDirectEditingFeature.java
index a338bd8..384c6b8 100644
--- a/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/SketchTextProposalDirectEditingFeature.java
+++ b/tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/SketchTextProposalDirectEditingFeature.java
@@ -46,8 +46,13 @@
 		if (proposalSupport == null) {
 			proposalSupport = new IProposalSupport() {
 
-				public void setValue(IProposal value, IDirectEditingContext context) {
-					SketchUtil.setCurrentLabelValue(context.getPictogramElement(), value.getText());
+				@Override
+				public void setValue(String value, IProposal proposal, IDirectEditingContext context) {
+					String text = value;
+					if (proposal != null && proposal.getText() != null) {
+						text = proposal.getText();
+					}
+					SketchUtil.setCurrentLabelValue(context.getPictogramElement(), text);
 				}
 
 				@Override