fixed little bugs