diff --git a/hif/inc/hif.h b/hif/inc/hif.h
index edc7699e19..c3e142bef6 100644
--- a/hif/inc/hif.h
+++ b/hif/inc/hif.h
@@ -939,7 +939,7 @@ ssize_t hif_dump_desc_event(struct hif_softc *scn, char *buf);
  * Return: void
  */
 void hif_set_ce_service_max_yield_time(struct hif_opaque_softc *hif,
-				       uint8_t ce_service_max_yield_time);
+				       uint32_t ce_service_max_yield_time);
 
 /**
  * hif_get_ce_service_max_yield_time() - get CE service max yield time
diff --git a/hif/src/hif_main.c b/hif/src/hif_main.c
index 715d74581f..845b2232a3 100644
--- a/hif/src/hif_main.c
+++ b/hif/src/hif_main.c
@@ -1261,12 +1261,12 @@ void hif_set_initial_wakeup_cb(struct hif_opaque_softc *hif_ctx,
 }
 
 void hif_set_ce_service_max_yield_time(struct hif_opaque_softc *hif,
-				       uint8_t ce_service_max_yield_time)
+				       uint32_t ce_service_max_yield_time)
 {
 	struct hif_softc *hif_ctx = HIF_GET_SOFTC(hif);
 
 	hif_ctx->ce_service_max_yield_time =
-		ce_service_max_yield_time * 1000 * 1000;
+		ce_service_max_yield_time * 1000;
 }
 
 unsigned long long