From edb8cfff0a8b734493bf8d3ec1e9977fb7418956 Mon Sep 17 00:00:00 2001
From: Shriansh Chari <30420527+shrianshChari@users.noreply.github.com>
Date: Mon, 26 Aug 2024 17:33:18 -0400
Subject: [PATCH] Clean up `legacy` check

---
 stats/src/classifier.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/stats/src/classifier.ts b/stats/src/classifier.ts
index 5f79639..f8f1f2a 100644
--- a/stats/src/classifier.ts
+++ b/stats/src/classifier.ts
@@ -255,8 +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 && ((legacy && DRAGONS.has(pokemon.species)) ||
-      (!legacy && gen.species.get(pokemon.species)?.types.includes('Dragon')))) {
+    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')) {