Small mistakes fix