fix type error

Signed-off-by: Haohua Li <lihaohua90@gmail.com>