diff --git a/esp8266_deauther/DisplayUI.cpp b/esp8266_deauther/DisplayUI.cpp
index 0d40786..5321166 100644
--- a/esp8266_deauther/DisplayUI.cpp
+++ b/esp8266_deauther/DisplayUI.cpp
@@ -461,7 +461,7 @@ void DisplayUI::update() {
 
     draw();
 
-    uint32_t timeout = settings.getAttackSettings().timeout * 1000;
+    uint32_t timeout = settings.getDisplaySettings().timeout * 1000;
 
     if (currentTime > timeout) {
         if (!tempOff) {
@@ -889,4 +889,4 @@ void DisplayUI::setTime(int h, int m, int s) {
     clockHour   = h;
     clockMinute = m;
     clockSecond = s;
-}
\ No newline at end of file
+}