diff --git a/busManager/coord/email_helpers.py b/busManager/coord/email_helpers.py index fdc18d7..671571d 100644 --- a/busManager/coord/email_helpers.py +++ b/busManager/coord/email_helpers.py @@ -27,8 +27,9 @@ def email_companies_bus_roll(request, query_set=None): if not company.contact_email: continue company_route = [] + for route in context.get("routes"): - if route.get("bus").company == company: + if route.get("company") == company: company_route.append(route) if not company_route: continue