fix build error

Signed-off-by: Zang MingJie <zealot0630@gmail.com>