fix  some code