fix code bug

fix code bug