diff --git a/stats/src/classifier.ts b/stats/src/classifier.ts
index 5a0eca1..f8f1f2a 100644
--- a/stats/src/classifier.ts
+++ b/stats/src/classifier.ts
@@ -255,7 +255,8 @@ function tag(gen: Generation, team: Array<PokemonSet<ID>>, stalliness: number, l
       pokemon.moves.some((m: ID) => ['block', 'meanlook', 'spiderweb'].includes(m))) {
       style.trappers++;
     }
-    if (style.dragons < 2 && DRAGONS.has(pokemon.species)) {
+    if (style.dragons < 2 && legacy ? DRAGONS.has(pokemon.species)
+      : gen.species.get(pokemon.species)?.types.includes('Dragon')) {
       style.dragons++;
     }
     if ((style.clearance < 2 && pokemon.ability === 'magicbounce') || moves.has('rapidspin')) {