repair a bug

repair a bug