Wrong if() fixed