Fix little changes