1 files changed, 4 insertions, 1 deletions
diff --git a/os/kernel/src/chmsg.c b/os/kernel/src/chmsg.c
index 5030ac809..1e89aa0a3 100644
--- a/os/kernel/src/chmsg.c
+++ b/os/kernel/src/chmsg.c
@@ -77,8 +77,11 @@ msg_t chMsgWait(void) {
   chSysLock();
   if (!chMsgIsPendingI(currp))
     chSchGoSleepS(THD_STATE_WTMSG);
-/*  msg = chMsgGetI(currp);*/
+#if defined(CH_ARCHITECTURE_STM8)
   msg = chMsgGetI((volatile Thread *)currp); /* Temporary hack.*/
+#else
+  msg = chMsgGetI(currp);
+#endif
   chSysUnlock();
   return msg;
 }