fix Issue #10

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