fixed #13.

Signed-off-by: Yimin <zhongym8903@gmail.com>