fix the bugs...