diff --git a/src/pogo_scaled_estimators/pokebattler_proxy.py b/src/pogo_scaled_estimators/pokebattler_proxy.py index 031fd81..d1317bf 100644 --- a/src/pogo_scaled_estimators/pokebattler_proxy.py +++ b/src/pogo_scaled_estimators/pokebattler_proxy.py @@ -166,4 +166,4 @@ class PokebattlerProxy: return next(filter(lambda mon: mon["pokemonId"] == name, self.pokemon))["type"] def move_type(self, name: str) -> str: - return next(filter(lambda move: move["moveId"] == name, self.moves))["type"] + return next(filter(lambda move: "moveId" in move and move["moveId"] == name, self.moves))["type"]