From 170dddfd7174ed3320571b54fc751d1d3445b128 Mon Sep 17 00:00:00 2001
From: Yuri D'Elia <wavexx@thregr.org>
Date: Sat, 20 Aug 2022 16:04:29 +0200
Subject: [PATCH] Set/clear loading_flag also during unload

---
 Firmware/ultralcd.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/Firmware/ultralcd.cpp b/Firmware/ultralcd.cpp
index 25a193f3a..e193a30f7 100644
--- a/Firmware/ultralcd.cpp
+++ b/Firmware/ultralcd.cpp
@@ -1867,7 +1867,10 @@ void mFilamentItem(uint16_t nTemp, uint16_t nTempBed)
                     if (eFilamentAction == FilamentAction::AutoLoad) eFilamentAction = FilamentAction::None; // i.e. non-autoLoad
                 }
                 if (eFilamentAction == FilamentAction::UnLoad)
-                enquecommand_P(MSG_M702); // unload filament
+                {
+                    loading_flag = true;
+                    enquecommand_P(MSG_M702); // unload filament
+                }
             }
             break;
         case FilamentAction::MmuLoad:
@@ -4886,6 +4889,7 @@ void unload_filament(float unloadLength)
 	lcd_setstatuspgm(MSG_WELCOME);
 	custom_message_type = CustomMsg::Status;
 	eFilamentAction = FilamentAction::None;
+	loading_flag = false;
 }
 
 /// @brief Set print fan speed