diff --git a/keyboards/bastardkb/tbk/tbk.h b/keyboards/bastardkb/tbk/tbk.h
index 3f9cd9915d..5958097880 100644
--- a/keyboards/bastardkb/tbk/tbk.h
+++ b/keyboards/bastardkb/tbk/tbk.h
@@ -15,9 +15,10 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-
 #pragma once
 
+#include "quantum.h"
+
 // clang-format off
 #define LAYOUT_split_4x6_5( \
     k00, k01, k02, k03, k04, k05,            k55, k54, k53, k52, k51, k50, \
diff --git a/lib/python/qmk/info.py b/lib/python/qmk/info.py
index 7460d84ad3..98505c1c8e 100644
--- a/lib/python/qmk/info.py
+++ b/lib/python/qmk/info.py
@@ -773,7 +773,7 @@ def avr_processor_rules(info_data, rules):
     """
     info_data['processor_type'] = 'avr'
     info_data['platform'] = rules['ARCH'] if 'ARCH' in rules else 'unknown'
-    info_data['protocol'] = 'V-USB' if rules.get('MCU') in VUSB_PROCESSORS else 'LUFA'
+    info_data['protocol'] = 'V-USB' if info_data['processor'] in VUSB_PROCESSORS else 'LUFA'
 
     if 'bootloader' not in info_data:
         info_data['bootloader'] = 'atmel-dfu'