Solving little mistake