Small if condition change