fix some small bugs .