diff --git a/rota_generator/shifts.py b/rota_generator/shifts.py index 70927c1..586a299 100644 --- a/rota_generator/shifts.py +++ b/rota_generator/shifts.py @@ -5257,8 +5257,10 @@ class RotaBuilder(object): for worker in self.get_all_workers(): if worker.site != current_site: try: + site = worker.site + site_count = len(self.workers_at_sites.get(site, [])) timetable.append( - f"