fix one bug

fix one bug