fix little bugs