diff --git a/src/register_types.cpp b/src/register_types.cpp
index 0beb221..2d4636a 100644
--- a/src/register_types.cpp
+++ b/src/register_types.cpp
@@ -1,8 +1,8 @@
 #include "register_types.h"
 #include "action.hpp"
 #include "character_actor.hpp"
+#include "character_awareness.hpp"
 #include "character_data.hpp"
-#include "enemy.hpp"
 #include "global_world_state.hpp"
 #include "goal_marker.hpp"
 #include "health.hpp"
@@ -45,7 +45,6 @@ void initialize_gdextension_types(ModuleInitializationLevel p_level)
     ClassDB::register_class<TunnelsGameState>();
     ClassDB::register_class<TunnelsPlayer>();
 
-    ClassDB::register_class<Enemy>();
     ClassDB::register_class<Health>();
     ClassDB::register_class<CharacterActor>();
     ClassDB::register_class<ProjectilePool>();
@@ -67,6 +66,7 @@ void initialize_gdextension_types(ModuleInitializationLevel p_level)
     ClassDB::register_class<goap::Planner>();
     ClassDB::register_class<GoalMarker>();
     ClassDB::register_class<LineGoalMarker>();
+    ClassDB::register_class<CharacterAwareness>();
 }
 
 extern "C"