fix a mini bugs.