diff --git a/behaviour_action.cpp b/behaviour_action.cpp
index b517d27..0351a74 100644
--- a/behaviour_action.cpp
+++ b/behaviour_action.cpp
@@ -1,4 +1,5 @@
 #include "behaviour_action.h"
+#include "core/object/class_db.h"
 #include "core/object/object.h"
 
 void BehaviourAction::_bind_methods() {
diff --git a/behaviour_composite.cpp b/behaviour_composite.cpp
index 009e670..d6d8109 100644
--- a/behaviour_composite.cpp
+++ b/behaviour_composite.cpp
@@ -1,4 +1,5 @@
 #include "behaviour_composite.h"
+#include "core/variant/typed_array.h"
 
 void BehaviourComposite::_bind_methods() {}
 
diff --git a/behaviour_node.cpp b/behaviour_node.cpp
index 642dceb..3984aa2 100644
--- a/behaviour_node.cpp
+++ b/behaviour_node.cpp
@@ -1,6 +1,7 @@
 #include "behaviour_node.h"
 #include "behaviour_nodes/behaviour_tree.h"
 #include "core/config/engine.h"
+#include "core/object/class_db.h"
 
 void BehaviourNode::_bind_methods() {
 	BIND_ENUM_CONSTANT(Fail);
diff --git a/behaviour_tree.cpp b/behaviour_tree.cpp
index 7807183..5b94483 100644
--- a/behaviour_tree.cpp
+++ b/behaviour_tree.cpp
@@ -1,7 +1,8 @@
 #include "behaviour_tree.h"
+#include "behaviour_nodes/behaviour_composite.h"
 #include "behaviour_nodes/behaviour_node.h"
-#include "behaviour_nodes/control_nodes.h"
 #include "core/config/engine.h"
+#include "core/variant/typed_array.h"
 
 void BehaviourTree::_bind_methods() {}