diff --git a/repository/OpenPonk-FSM/OPFsmStateController.class.st b/repository/OpenPonk-FSM/OPFsmStateController.class.st
index 6c06e8c..4743e15 100644
--- a/repository/OpenPonk-FSM/OPFsmStateController.class.st
+++ b/repository/OpenPonk-FSM/OPFsmStateController.class.st
@@ -18,7 +18,7 @@ OPFsmStateController >> buildEditorForm: aForm [
 		state: self model isFinal;
 		whenChangedDo: [ :newValue |
 			self model isFinal: newValue.
-			self diagramElement modelChanged ]
+			self modelChanged ]
 ]
 
 { #category : 'construction' }
diff --git a/repository/OpenPonk-FSM/OPFsmStateControllerTest.class.st b/repository/OpenPonk-FSM/OPFsmStateControllerTest.class.st
index a8d1df8..4fa483e 100644
--- a/repository/OpenPonk-FSM/OPFsmStateControllerTest.class.st
+++ b/repository/OpenPonk-FSM/OPFsmStateControllerTest.class.st
@@ -41,8 +41,8 @@ OPFsmStateControllerTest >> testFigureRefresh [
 		renderDiagramElement.
 	self assert: controller diagramElement modelElement name equals: ''.
 	controller model name: 'new name'.
-	controller diagramElement modelChanged.
+	controller modelChanged.
 	self assert:
-		(controller diagramElement canvas nodes anySatisfy: [ :any | 
+		(controller diagramElement canvas nodes anySatisfy: [ :any |
 			 (any isKindOf: RSLabel) and: [ any text = 'new name' ] ])
 ]