1 files changed, 8 insertions, 3 deletions

diff --git a/tmk_core/common/wait.h b/tmk_core/common/wait.h
index bdcb3f2a4..553f5243f 100644
--- a/tmk_core/common/wait.h
+++ b/tmk_core/common/wait.h
@@ -11,10 +11,15 @@ extern "C" {
 #   include <util/delay.h>
 #   define wait_ms(ms)  _delay_ms(ms)
 #   define wait_us(us)  _delay_us(us)
-#elif defined(PROTOCOL_CHIBIOS)
+#elif defined PROTOCOL_CHIBIOS
 #   include "ch.h"
-#   define wait_ms(ms) chThdSleepMilliseconds(ms)
-#   define wait_us(us) chThdSleepMicroseconds(us)
+#   if defined(STM32F3xx_MCUCONF)
+#       define wait_ms(ms) chSysPolledDelayX(MS2RTC(STM32_HCLK, (ms)))
+#       define wait_us(us) chSysPolledDelayX(US2RTC(STM32_HCLK, (us)))
+#   else
+#       define wait_ms(ms) chThdSleepMilliseconds(ms)
+#       define wait_us(us) chThdSleepMicroseconds(us)
+#   endif
 #elif defined(__arm__)
 #   include "wait_api.h"
 #else  // Unit tests