add fixed a bug