Fix a bug

fix a bug