fix  bugs