diff --git a/whalrus/ballots/ballot_levels.py b/whalrus/ballots/ballot_levels.py index 0139645..c71030e 100644 --- a/whalrus/ballots/ballot_levels.py +++ b/whalrus/ballots/ballot_levels.py @@ -99,8 +99,9 @@ def as_dict(self) -> NiceDict: @cached_property def as_weak_order(self) -> list: if isinstance(self.scale, ScaleFromList): - return [NiceSet(k for k in self.as_dict.keys() if self.as_dict[k] == list(self.as_dict.values())[v]) - for v in self.scale.argsort(list(set(self.as_dict.values())), reverse=True)] + present_val = list(set(self.as_dict.values())) + return [NiceSet(k for k in self.as_dict.keys() if self.as_dict[k] == present_val[v]) + for v in self.scale.argsort(present_val, reverse = True)] return [NiceSet(k for k in self.as_dict.keys() if self.as_dict[k] == v) for v in sorted(set(self.as_dict.values()), reverse=True)]