From 8eef1eaa7cdcd8f9672a93dc5e8f9771a1ce9a3d Mon Sep 17 00:00:00 2001
From: Dan Albert <dan@gingerhq.net>
Date: Sun, 15 Nov 2020 23:59:20 -0800
Subject: [PATCH] Fix mypy issue.

---
 gen/flights/traveltime.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gen/flights/traveltime.py b/gen/flights/traveltime.py
index 2a75afe3..4756d287 100644
--- a/gen/flights/traveltime.py
+++ b/gen/flights/traveltime.py
@@ -133,6 +133,9 @@ class TotEstimator:
                 tot = self.package.time_over_target
             else:
                 tot = flight.flight_plan.tot_for_waypoint(tot_waypoint)
+                if tot is None:
+                    logging.error(f"TOT waypoint for {flight} has no TOT")
+                    tot = self.package.time_over_target
         return tot - travel_time - self.HOLD_TIME
 
     def earliest_tot(self) -> timedelta: