fix small bugs.