diff --git a/examples/companion_radio/ui-new/UITask.cpp b/examples/companion_radio/ui-new/UITask.cpp
index 0cebc100..7821af3e 100644
--- a/examples/companion_radio/ui-new/UITask.cpp
+++ b/examples/companion_radio/ui-new/UITask.cpp
@@ -840,9 +840,7 @@ void UITask::loop() {
           _display->drawTextCentered(_display->width() / 2, 20, "Low Battery.");
           _display->drawTextCentered(_display->width() / 2, 40, "Shutting Down!");
           _display->endFrame();
-          #if !defined(THINKNODE_M1) && !defined(LILYGO_TECHO) // TODO: refactor eink variants to use EINK_DISPLAY macros to gate this properly
-          delay(3000);
-          #endif
+          if (_display->isEink() == false) { delay(3000); }
         }
         shutdown();
       }
diff --git a/examples/companion_radio/ui-tiny/UITask.cpp b/examples/companion_radio/ui-tiny/UITask.cpp
index 125c4f7b..bfaddf1d 100644
--- a/examples/companion_radio/ui-tiny/UITask.cpp
+++ b/examples/companion_radio/ui-tiny/UITask.cpp
@@ -735,7 +735,7 @@ void UITask::loop() {
         _display->setTextSize(1);
         _display->drawTextCentered(_display->width() / 2, 18, "Shutting down!");
         _display->endFrame();
-        delay(3000); // TODO: refactor eink variants to use EINK_DISPLAY macros to gate this properly
+        if (_display->isEink() == false) { delay(3000); }
         }
         shutdown();
       }