bug fixed

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