diff --git a/quantum/quantum.c b/quantum/quantum.c
index 585692d4a..ad957a1b1 100644
--- a/quantum/quantum.c
+++ b/quantum/quantum.c
@@ -1,5 +1,5 @@
 #include "quantum.h"
-#if defined(__AVR__)
+#ifdef PROTOCOL_LUFA
 #include "outputselect.h"
 #endif
 
@@ -215,7 +215,7 @@ bool process_record_quantum(keyrecord_t *record) {
 	  return false;
       break;
 	#endif
-    #if defined(__AVR__)
+    #ifdef PROTOCOL_LUFA
     case OUT_AUTO:
       if (record->event.pressed) {
         set_output(OUTPUT_AUTO);