correct small bugs