fix a small bugs.