Adjustment the param.

Signed-off-by: boddmg <boddmg@gmail.com>