diff --git a/register_types.cpp b/register_types.cpp
index 42e220f..d74a015 100644
--- a/register_types.cpp
+++ b/register_types.cpp
@@ -128,8 +128,10 @@ static LimboUtility *_limbo_utility = nullptr;
 
 void initialize_limboai_module(ModuleInitializationLevel p_level) {
 	if (p_level == MODULE_INITIALIZATION_LEVEL_SCENE) {
-		GDREGISTER_CLASS(BehaviorTreeData);
+#ifdef TOOLS_ENABLED
 		GDREGISTER_CLASS(BehaviorTreeView);
+#endif // TOOLS_ENABLED
+		GDREGISTER_CLASS(BehaviorTreeData);
 #ifdef LIMBOAI_GDEXTENSION
 		GDREGISTER_CLASS(LimboDebugger);
 #endif