diff --git a/repository/OpenPonk-FSM/OPFsmLayouterTest.class.st b/repository/OpenPonk-FSM/OPFsmLayouterTest.class.st
index 9f60a7f..89d31f1 100644
--- a/repository/OpenPonk-FSM/OPFsmLayouterTest.class.st
+++ b/repository/OpenPonk-FSM/OPFsmLayouterTest.class.st
@@ -14,6 +14,7 @@ OPFsmLayouterTest >> setUp [
 
 	| stateController1 stateController2 transitionController canvasPresenter model stateModel1 stateModel2 transitionModel |
 	diagramController := OPFsmController new.
+	diagramController diagramElement: OPDiagram new.
 	stateController1 := OPFsmStateController new.
 	stateController2 := OPFsmStateController new.
 	transitionController := OPFsmTransitionController new.