Move this line inside the if?