small fix
Signed-off-by: Li Mi <limisky@gmail.com>