diff --git a/hsm/limbo_hsm.cpp b/hsm/limbo_hsm.cpp
index 1fde7ae..09b2bcd 100644
--- a/hsm/limbo_hsm.cpp
+++ b/hsm/limbo_hsm.cpp
@@ -126,7 +126,7 @@ LimboState *LimboHSM::get_leaf_state() const {
 	}
 }
 
-void LimboHSM::set_initial_state(Node *p_state) {
+void LimboHSM::set_initial_state(LimboState *p_state) {
 	ERR_FAIL_COND(p_state == nullptr || !p_state->is_class("LimboState"));
 	initial_state = Object::cast_to<LimboState>(p_state);
 }
diff --git a/hsm/limbo_hsm.h b/hsm/limbo_hsm.h
index 6de82e2..f51eafa 100644
--- a/hsm/limbo_hsm.h
+++ b/hsm/limbo_hsm.h
@@ -60,7 +60,7 @@ public:
 	LimboState *get_leaf_state() const;
 	void set_active(bool p_active);
 
-	void set_initial_state(Node *p_state);
+	void set_initial_state(LimboState *p_state);
 	LimboState *get_initial_state() const { return initial_state; }
 
 	virtual void initialize(Node *p_agent, const Ref<Blackboard> &p_parent_scope = nullptr);