From 455cb9da99aabf13790bc1b01a2b049901e7678a Mon Sep 17 00:00:00 2001
From: alfrix <alfredomonclus@gmail.com>
Date: Tue, 2 Aug 2022 08:57:16 -0300
Subject: [PATCH] temperature: fix cooldown after 207914a

---
 panels/temperature.py | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/panels/temperature.py b/panels/temperature.py
index 6bfd3b13..aef87e33 100644
--- a/panels/temperature.py
+++ b/panels/temperature.py
@@ -216,15 +216,16 @@ class TemperaturePanel(ScreenPanel):
                 target = None
                 max_temp = float(self._printer.get_config_section(heater)['max_temp'])
                 name = heater.split()[1] if len(heater.split()) > 1 else heater
-                for i in self.preheat_options[setting]:
-                    logging.info(f"{self.preheat_options[setting]}")
-                    if i == name:
-                        # Assign the specific target if available
-                        target = self.preheat_options[setting][name]
-                        logging.info(f"name match {name}")
-                    elif i == heater:
-                        target = self.preheat_options[setting][heater]
-                        logging.info(f"heater match {heater}")
+                with contextlib.suppress(KeyError):
+                    for i in self.preheat_options[setting]:
+                        logging.info(f"{self.preheat_options[setting]}")
+                        if i == name:
+                            # Assign the specific target if available
+                            target = self.preheat_options[setting][name]
+                            logging.info(f"name match {name}")
+                        elif i == heater:
+                            target = self.preheat_options[setting][heater]
+                            logging.info(f"heater match {heater}")
                 if target is None and setting == "cooldown":
                     target = 0
                 if heater.startswith('extruder'):