diff --git a/src/wx/DfuPanel.cpp b/src/wx/DfuPanel.cpp
index 267e793..73bd764 100644
--- a/src/wx/DfuPanel.cpp
+++ b/src/wx/DfuPanel.cpp
@@ -167,7 +167,7 @@ void DfuPanel::SetDeviceState(const DeviceState& state)
         GetMainFrame()->ShowMain();
     }
 
-    if (!state.connected)
+    if (!m_isEnteringDfu && !state.connected)
     {
         m_isEnteringDfu = false;
         m_backButton->Enable();