fix Issue #18

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