Add a fix.

Signed-off-by: windyan <yanhaifeng@gmail.com>
Merge pull request #1 from windyan/contrib/windyan/test

Add a fix.