fixing little bugs