fix two new little bugs