fix bug

Signed-off-by: Cheng Li <scrappedprince.li@gmail.com>