diff --git a/src/utility/wifi_drv.cpp b/src/utility/wifi_drv.cpp
index a0920077..5d620720 100644
--- a/src/utility/wifi_drv.cpp
+++ b/src/utility/wifi_drv.cpp
@@ -1060,7 +1060,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 c20db43b..a91824fa 100644
--- a/src/utility/wifi_drv.h
+++ b/src/utility/wifi_drv.h
@@ -290,7 +290,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);