fix  bug