diff --git a/src/mbtaclient/handlers/base_handler.py b/src/mbtaclient/handlers/base_handler.py index 71fcd49..e2f9332 100644 --- a/src/mbtaclient/handlers/base_handler.py +++ b/src/mbtaclient/handlers/base_handler.py @@ -407,8 +407,8 @@ def check_route_level_alert(informed_entity: MBTAAlertsInformedEntity): for informed_entity in mbta_alert.informed_entities: - active_period_start = mbta_alert.active_period_start.replace(tzinfo=None) - active_period_end = mbta_alert.active_period_end.replace(tzinfo=None) + active_period_start = mbta_alert.active_period_start.replace(tzinfo=None) if mbta_alert.active_period_start else None + active_period_end = mbta_alert.active_period_end.replace(tzinfo=None) if mbta_alert.active_period_end else None try: if (