The function has a little fix during upstream review, do fix to sync with upstream. Change-Id: I9e1c43a660b2f83395d1639aa962988ca04494e5 Signed-off-by: Jianqun Xu <jay.xu@rock-chips.com>