Fix #120

@Harry-Chen Is this working?
Signed-off-by: Peter Cai <xqsx43cxy@126.com>