diff --git a/src/utility/wifi_drv.cpp b/src/utility/wifi_drv.cpp
index c16656f6..a8b07e34 100644
--- a/src/utility/wifi_drv.cpp
+++ b/src/utility/wifi_drv.cpp
@@ -1093,7 +1093,11 @@ void WiFiDrv::pinMode(uint8_t pin, uint8_t mode)
     SpiDrv::spiSlaveDeselect();
 }
 
+#if defined(ARDUINO_ARCH_MBED)
 PinStatus WiFiDrv::digitalRead(uint8_t pin)
+#else
+int WiFiDrv::digitalRead(uint8_t pin)
+#endif
 {
     WAIT_FOR_SLAVE_SELECT();
     // Send Command
diff --git a/src/utility/wifi_drv.h b/src/utility/wifi_drv.h
index 2d1872f4..d444d3e6 100644
--- a/src/utility/wifi_drv.h
+++ b/src/utility/wifi_drv.h
@@ -297,7 +297,13 @@ class WiFiDrv
     static void debug(uint8_t on);
     static float getTemperature();
     static void pinMode(uint8_t pin, uint8_t mode);
+    
+#if defined(ARDUINO_ARCH_MBED)
     static PinStatus digitalRead(uint8_t pin);
+#else    
+    static int digitalRead(uint8_t pin);
+#endif    
+    
     static void digitalWrite(uint8_t pin, uint8_t value);
     static uint16_t analogRead(uint8_t adc_channel);
     static void analogWrite(uint8_t pin, uint8_t value);