diff --git a/libs/main/state_machine/pistatemachine.h b/libs/main/state_machine/pistatemachine.h
index 51a52f06..01b43893 100644
--- a/libs/main/state_machine/pistatemachine.h
+++ b/libs/main/state_machine/pistatemachine.h
@@ -40,7 +40,7 @@ public:
 
 	bool start();
 	bool isRunning() const { return is_running; }
-	void addOnFinish(std::function<void()> f) { on_finish = f; }
+	void setOnFinish(std::function<void()> f) { on_finish = f; }
 
 	template<typename... Args>
 	bool postEvent(int event_id, Args... args) {
diff --git a/main.cpp b/main.cpp
index 35c1849d..c748f73f 100644
--- a/main.cpp
+++ b/main.cpp
@@ -100,7 +100,7 @@ int main(int argc, char * argv[]) {
 
 	bool posted;
 	PIStateMachine * root = new PIStateMachine("Machine");
-	root->addOnFinish([] { piCout << "finish"; });
+	root->setOnFinish([] { piCout << "finish"; });
 
 	PIStateLambda * s1   = new PIStateLambda([] { piCout << "+ enter s1"; }, [] { piCout << "- exit s1"; }, "s1");
 	PIStateLambda * s2   = new PIStateLambda([] { piCout << "+ enter s2"; }, [] { piCout << "- exit s2"; }, "s2");