fixed two little bugs