fix some code bug.