diff --git a/src/ezButton.cpp b/src/ezButton.cpp
index 2c6b618..31e6d5b 100644
--- a/src/ezButton.cpp
+++ b/src/ezButton.cpp
@@ -39,7 +39,11 @@ ezButton::ezButton(int pin, int mode) {
 	count = 0;
 	countMode = COUNT_FALLING;
 
-	pinMode(btnPin, mode);
+	#if defined(ARDUINO_ARCH_STM32F1)
+		pinMode(btnPin, (WiringPinMode)mode);
+	#else
+		pinMode(btnPin, mode);
+	#endif
 
 	previousSteadyState = digitalRead(btnPin);
 	lastSteadyState = previousSteadyState;