diff --git a/postcode_lookup/template_sorter.py b/postcode_lookup/template_sorter.py index eb08cc6..28d7235 100644 --- a/postcode_lookup/template_sorter.py +++ b/postcode_lookup/template_sorter.py @@ -177,12 +177,8 @@ def __init__(self, *args, **kwargs) -> None: @property def weight(self): - if self.timetable.is_before(TimetableEvent.REGISTRATION_DEADLINE): - return -5999 - - if self.timetable.is_after(TimetableEvent.REGISTRATION_DEADLINE): - return 1002 - return 0 + parent_weight = super().weight + return 0 if parent_weight == 0 else parent_weight + 1 @property def context(self):