fix a bug

Signed-off-by: Yifan Ying <nafan628@gmail.com>