Fixes #2

Signed-off-by: Bao Haojun <baohaojun@gmail.com>