fix function bug

fix function bug