Bug fix:                   fix #5 issue.

Signed-off-by: Euhow <euhowzy@gmail.com>