diff --git a/src/register_types.cpp b/src/register_types.cpp
index f54187a..92d2518 100644
--- a/src/register_types.cpp
+++ b/src/register_types.cpp
@@ -4,7 +4,10 @@
 #include "health.hpp"
 #include "pellet_projectile.hpp"
 #include "player_character.hpp"
+#include "player_states.hpp"
 #include "projectile_pool.hpp"
+#include "state.hpp"
+#include "state_machine.hpp"
 #include "tunnels_game_mode.hpp"
 #include "tunnels_game_state.hpp"
 #include "tunnels_player.hpp"
@@ -51,6 +54,10 @@ void initialize_gdextension_types(ModuleInitializationLevel p_level)
 
     ClassDB::register_class<Projectile>();
     ClassDB::register_class<PelletProjectile>();
+
+    ClassDB::register_class<StateMachine>();
+    ClassDB::register_class<PlayerState>();
+    ClassDB::register_class<FollowingPlayer>();
 }
 
 extern "C"