fix build error

Signed-off-by: Zang MingJie <zealot0630@gmail.com>
Merge pull request #49 from kghost/patch-2

fix build error