fix some unittest