fix Issue #22

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