Fix #8

Signed-off-by: Yeechan Lu <wz.bluesnow@gmail.com>