fixed a bug !