diff --git a/modules/wave_survival/enemy_body.cpp b/modules/wave_survival/enemy_body.cpp
index b5b4d1c8..8bf9cbc6 100644
--- a/modules/wave_survival/enemy_body.cpp
+++ b/modules/wave_survival/enemy_body.cpp
@@ -4,6 +4,7 @@
 
 void EnemyBody::_bind_methods() {
 	ClassDB::bind_method(D_METHOD("set_movement_direction", "direction"), &self_type::set_movement_direction);
+	ClassDB::bind_method(D_METHOD("get_unit"), &self_type::get_unit);
 }
 
 void EnemyBody::on_child_added(Node *node) {