Fix little mistake