Small fix.

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