fix little issues