diff --git a/include/battle_z_move.h b/include/battle_z_move.h
index 5d34e25a92..2d1ff1a006 100644
--- a/include/battle_z_move.h
+++ b/include/battle_z_move.h
@@ -13,6 +13,7 @@ struct SignatureZMove
     u16 zmove;
 };
 
+bool8 IsZMove(u16 move);
 void QueueZMove(u8 battlerId, u16 baseMove);
 bool32 IsViableZMove(u8 battlerId, u16 move);
 bool32 TryChangeZIndicator(u8 battlerId, u8 moveIndex);
diff --git a/src/battle_ai_main.c b/src/battle_ai_main.c
index f2b48af614..d3d158c769 100644
--- a/src/battle_ai_main.c
+++ b/src/battle_ai_main.c
@@ -6,6 +6,7 @@
 #include "battle_ai_main.h"
 #include "battle_factory.h"
 #include "battle_setup.h"
+#include "battle_z_move.h"
 #include "data.h"
 #include "event_data.h"
 #include "item.h"