add : fix the bugs