--Fix issue #4

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