repair a little bug

repair a little bug