diff --git a/drivers/misc/mediatek/connectivity/common/conn_soc/core/include/stp_core.h b/drivers/misc/mediatek/connectivity/common/conn_soc/core/include/stp_core.h
index eaa5ce773e332..9393a2acdb891 100644
--- a/drivers/misc/mediatek/connectivity/common/conn_soc/core/include/stp_core.h
+++ b/drivers/misc/mediatek/connectivity/common/conn_soc/core/include/stp_core.h
@@ -529,7 +529,7 @@ extern void mtk_wcn_stp_set_auto_rst(MTK_WCN_BOOL auto_rst);
 * RETURNS
 *  0: Sccuess  Negative value: Fail
 *****************************************************************************/
-extern int mtk_wcn_stp_psm_notify_stp(const UINT32 action);
+extern int mtk_wcn_stp_psm_notify_stp(const MTKSTP_PSM_ACTION_T action);
 
 extern int mtk_wcn_stp_set_psm_state(MTKSTP_PSM_STATE_T state);