diff --git a/src/dungeon_ai.c b/src/dungeon_ai.c
index 3d9f7b154..590849e9a 100644
--- a/src/dungeon_ai.c
+++ b/src/dungeon_ai.c
@@ -59,8 +59,8 @@ u32 sub_8075818(struct DungeonEntity *entity)
             {
                 case ENTITY_NONE:
                 case ENTITY_POKEMON:
-                case 4:
-                case 5:
+                case ENTITY_UNK_4:
+                case ENTITY_UNK_5:
                     break;
                 case ENTITY_TRAP:
                     trapData = (u8*) GetTrapData(subEntity);