diff --git a/src/Power.cpp b/src/Power.cpp
index 4e30c8288..ba908e3fb 100644
--- a/src/Power.cpp
+++ b/src/Power.cpp
@@ -136,7 +136,7 @@ bool Power::setup()
 
 void Power::shutdown()
 {
-#ifdef AXP192_SLAVE_ADDRESS
+#ifdef TBEAM_V10
     DEBUG_MSG("Shutting down\n");
     axp.shutdown();
 #endif
diff --git a/src/main.cpp b/src/main.cpp
index 228f2b3e3..4c0af7f7b 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -233,7 +233,7 @@ class ButtonThread : public OSThread
 
         // If user button is held down for 10 seconds, shutdown the device.
         if (millis() - longPressTime > 10 * 1000) {
-#ifdef AXP192_SLAVE_ADDRESS
+#ifdef TBEAM_V10
             if (axp192_found == true) {
                 power->shutdown();
             }