Change the if condition to be more specific