Add a fix.

Signed-off-by: windyan <yanhaifeng@gmail.com>