Attempt to fix Issue #1

First time to Coffee, not sure if it is right.

Signed-off-by: liuyanghejerry <liuyanghejerry@126.com>