From ff7fb3870b8ce830a778983375bc13c33ff5eb45 Mon Sep 17 00:00:00 2001
From: dexslab <dex35803@gmail.com>
Date: Wed, 23 Sep 2020 17:54:47 -0400
Subject: [PATCH] Fix for uno compile with wifi interface that is not needed
 and fix for compile on both motorshield type not being used properly

---
 CommandStation-EX.ino | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CommandStation-EX.ino b/CommandStation-EX.ino
index d9e16ac..f9557ef 100644
--- a/CommandStation-EX.ino
+++ b/CommandStation-EX.ino
@@ -16,7 +16,7 @@
 #include "DCC.h"
 #include "DIAG.h"
 #include "DCCEXParser.h"
-#if ENABLE_WIFI
+#if ENABLE_WIFI && (defined(ARDUINO_AVR_MEGA) || defined(ARDUINO_AVR_MEGA2560))
 #include "WifiInterface.h"
 #endif
 #if ENABLE_FREE_MEM_WARNING
@@ -146,7 +146,7 @@ void setup()
   // Optionally a Timer number (1..4) may be passed to DCC::begin to override the default Timer1 used for the
   // waveform generation.  e.g.  DCC::begin(STANDARD_MOTOR_SHIELD,2); to use timer 2
 
-  DCC::begin(MOTOR_BOARD);
+  DCC::begin(MOTOR_SHIELD_TYPE);
 }
 
 void loop()